| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_unified_experimental_test.cc | 374 TF_OutputList* add_outputs = TF_NewOutputList(); in TEST_P() local
375 TF_OutputListSetNumOutputs(add_outputs, 1, status.get()); in TEST_P()
379 TF_ExecuteOperation(add_op, 2, inputs, add_outputs, status.get()); in TEST_P()
386 TF_FinalizeFunction(graph_ctx, add_outputs, status.get()); in TEST_P()
390 TF_DeleteAbstractTensor(TF_OutputListGet(add_outputs, 0)); in TEST_P()
417 TF_ExecuteOperation(fn_op, 1, &input_t, add_outputs, status.get()); in TEST_P()
420 ASSERT_EQ(1, TF_OutputListNumOutputs(add_outputs)); in TEST_P()
421 TF_AbstractTensor* final_result = TF_OutputListGet(add_outputs, 0); in TEST_P()
430 TF_DeleteOutputList(add_outputs); in TEST_P()
571 TF_OutputList* add_outputs = TF_NewOutputList(); in TEST_P() local
[all …]
|
| D | mnist_gradients_testutil.cc | 52 std::vector<AbstractTensorHandle*> add_outputs(1); in AddGradModel() local
55 ops::Add(tape_ctx.get(), inputs, absl::MakeSpan(add_outputs), "Add")); in AddGradModel()
56 TF_RETURN_IF_ERROR(tape->ComputeGradient(ctx, /*targets=*/add_outputs, in AddGradModel()
59 for (auto add_output : add_outputs) { in AddGradModel()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | special_selector.cc | 154 auto add_outputs = graph.FindOutputs(add_node->id); in TryFCFCAdd() local
166 it = tensor_descriptors.find(add_outputs[0]->id); in TryFCFCAdd()
175 InitSingleOpSubgraph(fc0_inputs, add_outputs, gpu_subgraph); in TryFCFCAdd()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_test.py | 213 output, = custom.add_outputs(output)
243 return custom.add_outputs(a * x + b, x)
272 output, = custom.add_outputs(output)
316 custom.add_outputs([res])
|
| D | op_hint.py | 458 def add_outputs(self, *args, **kwargs): member in OpHint
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 112 auto output = op_info->add_outputs(); in DescribeArbitraryRankOutput()
309 DescribeTensor4D(size1, 1, 1, 1, op_context.op_info.add_outputs()); in DescribeUnaryOp()
328 DescribeTensor4D(2 * size1, size2, 1, 1, op_context.op_info.add_outputs()); in DescribeBinaryOp()
344 DescribeTensor4D(1, 1, size2, size1, op_context.op_info.add_outputs()); in DescribeBiasAdd()
442 DescribeTensor4D(y[0], y[1], y[2], y[3], op_info.add_outputs()); in DescribePoolingOp()
450 DescribeTensor4D(x[0], x[1], x[2], x[3], op_info.add_outputs()); in DescribePoolingOp()
456 DescribeTensor4D(x[0], x[1], x[2], x[3], op_info.add_outputs()); in DescribePoolingOp()
492 auto* tensor = op_info.add_outputs(); in DescribeFusedBatchNorm()
2108 DescribeTensor4D(0, 0, 0, 0, op_context.op_info.add_outputs()); in TEST_F()
2122 DescribeTensor4D(0, 0, 0, 0, op_context.op_info.add_outputs()); in TEST_F()
[all …]
|
| D | virtual_scheduler.cc | 767 *op_info.add_outputs() = output; in CreateOpContext()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.lite.-op-hint.pbtxt | 74 name: "add_outputs"
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 48 op_context.op_info.add_outputs()->Swap(&output); in PredictExecutionTime()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 230 add_outputs = (constant_op.constant(4.),)
235 outputs=add_outputs,
255 add_outputs = (constant_op.constant(4.),)
260 outputs=add_outputs,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1675 sig_def_builder.add_outputs(outputs_buf); in CreateSignatureDefs()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_generated.h | 10759 void add_outputs(flatbuffers::Offset<flatbuffers::Vector<int32_t>> outputs) {
10808 builder_.add_outputs(outputs);
10912 void add_outputs(flatbuffers::Offset<flatbuffers::Vector<int32_t>> outputs) {
10943 builder_.add_outputs(outputs);
11245 …void add_outputs(flatbuffers::Offset<flatbuffers::Vector<flatbuffers::Offset<tflite::TensorMap>>> …
11275 builder_.add_outputs(outputs);
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 1043 output = op->add_outputs(); in TEST()
|