Searched refs:ElementwiseOperation (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.cc | 117 std::string ElementwiseOperationString(ElementwiseOperation op) { in ElementwiseOperationString() 119 case ElementwiseOperation::kAdd: in ElementwiseOperationString() 121 case ElementwiseOperation::kMultiply: in ElementwiseOperationString()
|
D | dnn.h | 931 enum class ElementwiseOperation { kAdd, kMultiply }; enum 933 std::string ElementwiseOperationString(ElementwiseOperation op); 1930 Stream* stream, ElementwiseOperation operation, 1958 Stream* stream, ElementwiseOperation operation, in DoElementwiseOperateScaledQuantized()
|
D | stream.h | 666 dnn::ElementwiseOperation operation, 673 dnn::ElementwiseOperation operation,
|
D | stream.cc | 65 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/ |
D | cuda_dnn.h | 528 Stream* stream, dnn::ElementwiseOperation operation,
|
D | cuda_dnn.cc | 4579 Stream* stream, dnn::ElementwiseOperation operation, in DoElementwiseOperate()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 544 Stream* stream, dnn::ElementwiseOperation operation,
|
D | rocm_dnn.cc | 4539 Stream* stream, dnn::ElementwiseOperation operation, in DoElementwiseOperate()
|