| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | identify_l2_normalization_test.cc | 49 auto sum_op = new TensorFlowSumOperator; in RunIdentifyL2Normalization() local
50 sum_op->inputs = {sq_op->outputs[0]}; in RunIdentifyL2Normalization()
51 sum_op->outputs = {"Sumoutput"}; in RunIdentifyL2Normalization()
58 sqrt_op->inputs = {sum_op->outputs[0]}; in RunIdentifyL2Normalization()
71 model.operators.push_back(std::unique_ptr<Operator>(sum_op)); in RunIdentifyL2Normalization()
78 rsqrt_op->inputs = {sum_op->outputs[0]}; in RunIdentifyL2Normalization()
91 model.operators.push_back(std::unique_ptr<Operator>(sum_op)); in RunIdentifyL2Normalization()
|
| D | resolve_constant_unary_test.cc | 50 auto sum_op = absl::make_unique<TensorFlowSumOperator>(); in RunResolveSum() local
51 sum_op->keep_dims = true; in RunResolveSum()
52 sum_op->inputs = {"input0", "input1"}; in RunResolveSum()
53 sum_op->outputs = {output_name}; in RunResolveSum()
54 model.operators.push_back(std::move(sum_op)); in RunResolveSum()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_l2_normalization.cc | 101 Operator* sum_op = in Run() local
103 if (sum_op->type != OperatorType::kSum) { in Run()
107 LogName(*sum_op)); in Run()
111 Operator* square_op = GetOpWithOutput(*model, sum_op->inputs[0]); in Run()
140 DeleteOpAndArrays(model, sum_op); in Run()
|
| D | resolve_constant_unary.cc | 255 auto sum_op = static_cast<const TensorFlowSumOperator*>(unary_op); in Run() local
259 sum_op->keep_dims, axis_array.GetBuffer<ArrayDataType::kInt32>().data, in Run()
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | math_ops.cc | 120 AbstractOperationPtr sum_op(ctx->CreateOperation()); in Sum() local
121 TF_RETURN_IF_ERROR(sum_op->Reset("Sum", /*raw_device_name=*/nullptr)); in Sum()
122 TF_RETURN_IF_ERROR(MaybeSetOpName(sum_op.get(), name)); in Sum()
123 TF_RETURN_IF_ERROR(sum_op->AddInput(inputs[0])); // input_vals in Sum()
124 TF_RETURN_IF_ERROR(sum_op->AddInput(inputs[1])); // reduction_indices in Sum()
127 TF_RETURN_IF_ERROR(sum_op->Execute(outputs, &num_retvals)); in Sum()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorFunctors.h | 102 internal::scalar_sum_op<T> sum_op;
103 *accum = sum_op(*accum, t);
127 internal::scalar_sum_op<T> sum_op;
128 return sum_op(saccum, predux(vaccum));
150 internal::scalar_sum_op<T> sum_op;
151 *accum = sum_op(*accum, t);
177 internal::scalar_sum_op<T> sum_op;
178 …return sum_op(saccum, predux(vaccum)) / (scalarCount_ + packetCount_ * unpacket_traits<Packet>::si…
|
| /external/eigen/unsupported/test/ |
| D | cxx11_meta.cpp | 37 using Eigen::internal::sum_op;
318 VERIFY((array_zip<sum_op>(a, b) == c)); in test_array_zip_and_apply()
321 VERIFY_IS_EQUAL((array_apply_and_reduce<sum_op, times2_op>(a)), 216); in test_array_zip_and_apply()
322 VERIFY_IS_EQUAL((array_apply_and_reduce<sum_op, times2_op>(b)), 30); in test_array_zip_and_apply()
323 VERIFY_IS_EQUAL((array_zip_and_reduce<product_op, sum_op>(a, b)), 14755932); in test_array_zip_and_apply()
324 VERIFY_IS_EQUAL((array_zip_and_reduce<sum_op, product_op>(a, b)), 388); in test_array_zip_and_apply()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_aggregate_ops.cc | 239 mkldnn::sum sum_op(sum_pd); in Compute() local
245 sum_op.execute(*fwd_cpu_stream, net_args); in Compute()
|
| /external/eigen/unsupported/Eigen/CXX11/src/util/ |
| D | CXX11Meta.h | 291 struct sum_op {
329 constexpr inline decltype(reduce<sum_op, Ts...>::run((*((Ts*)0))...)) arg_sum(Ts... ts)
331 return reduce<sum_op, Ts...>::run(ts...);
391 …r inline auto array_sum(const array<T, N>& arr) -> decltype(array_reduce<sum_op, T, N>(arr, static…
393 return array_reduce<sum_op, T, N>(arr, static_cast<T>(0));
|
| D | EmulateCXX11Meta.h | 212 struct sum_op {
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_clusterspec_prop_test.py | 385 sum_op = mul0 + mul1 + mul2
395 sum_op,
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 923 tensorflow::NodeDef* sum_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local
924 sum_op->set_op("Sum"); in ConvertL2NormalizationOperator()
925 sum_op->set_name(sum_output); in ConvertL2NormalizationOperator()
926 *sum_op->add_input() = square_output; in ConvertL2NormalizationOperator()
927 *sum_op->add_input() = sum_reduction_indices; in ConvertL2NormalizationOperator()
928 (*sum_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2NormalizationOperator()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 393 auto sum_op = ops::Sum(scope.WithOpName("sum").WithDevice("/device:GPU:0"), in ReduceTransposerKeepDims() local
396 auto z = ops::Identity(scope.WithOpName("z"), sum_op); in ReduceTransposerKeepDims()
460 auto sum_op = ops::Max(scope.WithOpName("max").WithDevice("/device:GPU:0"), in ReduceTransposerValidAxisNode() local
463 auto z = ops::Identity(scope.WithOpName("z"), sum_op); in ReduceTransposerValidAxisNode()
|