diff --git a/ttnn/cpp/ttnn/operations/eltwise/unary/device/unary_composite_op.cpp b/ttnn/cpp/ttnn/operations/eltwise/unary/device/unary_composite_op.cpp index d87bad0f779..22f63e9fd2f 100644 --- a/ttnn/cpp/ttnn/operations/eltwise/unary/device/unary_composite_op.cpp +++ b/ttnn/cpp/ttnn/operations/eltwise/unary/device/unary_composite_op.cpp @@ -666,16 +666,16 @@ Tensor _polygamma(const Tensor& input_a, int32_t k, const std::optional& memory_config, std::optional optional_output_tensor) { +Tensor ExecuteRdiv::operator()(uint8_t queue_id, const Tensor& input_tensor, float value, const std::string& round_mode, const std::optional& memory_config, std::optional optional_output_tensor) { float t_inf = std::numeric_limits::infinity(); Tensor recip_result = ttnn::reciprocal(queue_id, input_tensor, memory_config, optional_output_tensor); Tensor result = ttnn::multiply(queue_id, recip_result, value, std::nullopt, memory_config, optional_output_tensor); if(round_mode == "trunc"){ - result = trunc(result); + result = ttnn::trunc(result); } else if(round_mode == "floor"){ - result = floor(result); + result = ttnn::floor(result); } return ttnn::where(ttnn::eqz(queue_id, input_tensor, memory_config), t_inf, result, memory_config, optional_output_tensor); } diff --git a/ttnn/cpp/ttnn/operations/eltwise/unary/unary_composite.hpp b/ttnn/cpp/ttnn/operations/eltwise/unary/unary_composite.hpp index b90979cb962..5b28723d046 100644 --- a/ttnn/cpp/ttnn/operations/eltwise/unary/unary_composite.hpp +++ b/ttnn/cpp/ttnn/operations/eltwise/unary/unary_composite.hpp @@ -119,7 +119,7 @@ struct ExecuteRdiv { uint8_t queue_id, const Tensor& input_tensor, float value, - string round_mode = "None", + const std::string& round_mode = "None", const std::optional& memory_config = std::nullopt, std::optional optional_output_tensor = std::nullopt); }; diff --git a/ttnn/cpp/ttnn/operations/eltwise/unary/unary_pybind.hpp b/ttnn/cpp/ttnn/operations/eltwise/unary/unary_pybind.hpp index 8fd65ab341d..dbb42407116 100644 --- a/ttnn/cpp/ttnn/operations/eltwise/unary/unary_pybind.hpp +++ b/ttnn/cpp/ttnn/operations/eltwise/unary/unary_pybind.hpp @@ -447,7 +447,7 @@ void bind_unary_rdiv(py::module& module, const unary_operation_t& operation, con [](const unary_operation_t& self, const ttnn::Tensor& input_tensor, float parameter_a, - string parameter_b, + const std::string& parameter_b, const std::optional& memory_config, const std::optional& output_tensor, const uint8_t& queue_id) {