Home
last modified time | relevance | path

Searched defs:sum_op (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_l2_normalization.cc101 Operator* sum_op = in Run() local
Dresolve_constant_unary.cc255 auto sum_op = static_cast<const TensorFlowSumOperator*>(unary_op); in Run() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dresolve_constant_unary_test.cc50 auto sum_op = absl::make_unique<TensorFlowSumOperator>(); in RunResolveSum() local
Didentify_l2_normalization_test.cc49 auto sum_op = new TensorFlowSumOperator; in RunIdentifyL2Normalization() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_aggregate_ops.cc239 mkldnn::sum sum_op(sum_pd); in Compute() local
/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc120 AbstractOperationPtr sum_op(ctx->CreateOperation()); in Sum() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc393 auto sum_op = ops::Sum(scope.WithOpName("sum").WithDevice("/device:GPU:0"), in ReduceTransposerKeepDims() local
460 auto sum_op = ops::Max(scope.WithOpName("max").WithDevice("/device:GPU:0"), in ReduceTransposerValidAxisNode() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc923 tensorflow::NodeDef* sum_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local