HomeSort by relevance Sort by last modified time
    Searched full:elementwiseoperation (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
dnn.cc 78 string ElementwiseOperationString(ElementwiseOperation op) {
80 case ElementwiseOperation::kAdd:
82 case ElementwiseOperation::kMultiply:
dnn.h 874 enum class ElementwiseOperation { kAdd, kMultiply };
876 string ElementwiseOperationString(ElementwiseOperation op);
    [all...]
stream.h     [all...]
stream.cc 66 string ToVlogString(dnn::ElementwiseOperation op) {
    [all...]
  /external/tensorflow/tensorflow/contrib/tensorrt/convert/
convert_nodes.cc 792 static const std::unordered_map<string, nvinfer1::ElementWiseOperation> ops{
793 {"Add", nvinfer1::ElementWiseOperation::kSUM},
794 {"Mul", nvinfer1::ElementWiseOperation::kPROD},
795 // {"max", nvinfer1::ElementWiseOperation::kMAX},
796 // {"min", nvinfer1::ElementWiseOperation::kMIN},
797 {"Sub", nvinfer1::ElementWiseOperation::kSUB},
798 {"Div", nvinfer1::ElementWiseOperation::kDIV},
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h 549 Stream* stream, dnn::ElementwiseOperation operation,
    [all...]
cuda_dnn.cc     [all...]

Completed in 74 milliseconds