Home
last modified time | relevance | path

Searched refs:ElementwiseOperation (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc117 std::string ElementwiseOperationString(ElementwiseOperation op) { in ElementwiseOperationString()
119 case ElementwiseOperation::kAdd: in ElementwiseOperationString()
121 case ElementwiseOperation::kMultiply: in ElementwiseOperationString()
Ddnn.h931 enum class ElementwiseOperation { kAdd, kMultiply }; enum
933 std::string ElementwiseOperationString(ElementwiseOperation op);
1930 Stream* stream, ElementwiseOperation operation,
1958 Stream* stream, ElementwiseOperation operation, in DoElementwiseOperateScaledQuantized()
Dstream.h666 dnn::ElementwiseOperation operation,
673 dnn::ElementwiseOperation operation,
Dstream.cc65 std::string ToVlogString(dnn::ElementwiseOperation op) { in ToVlogString()
1170 dnn::ElementwiseOperation operation, in ThenElementwiseOperate()
1189 dnn::ElementwiseOperation operation, in ThenElementwiseOperateScaledQuantized()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h528 Stream* stream, dnn::ElementwiseOperation operation,
Dcuda_dnn.cc4579 Stream* stream, dnn::ElementwiseOperation operation, in DoElementwiseOperate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h544 Stream* stream, dnn::ElementwiseOperation operation,
Drocm_dnn.cc4539 Stream* stream, dnn::ElementwiseOperation operation, in DoElementwiseOperate()