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()
|