| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 68 auto input = op_info->add_inputs(); in DescribeMatrix()
100 auto input = op_info->add_inputs(); in DescribeArbitraryRankInput()
208 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeConvolution()
209 DescribeTensor4D(kx, ky, iz2, oz, op_context.op_info.add_inputs()); in DescribeConvolution()
225 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeDepthwiseConv2dNative()
226 DescribeTensor4D(kx, ky, iz2, cm, op_context.op_info.add_inputs()); in DescribeDepthwiseConv2dNative()
256 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
258 DescribeTensor4D(batch, iz1, ix, iy, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
264 op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
267 DescribeTensor4D(kx, ky, iz2, oz, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
[all …]
|
| D | utils.cc | 271 *op_info.add_inputs() = input; in BuildOpInfoWithoutDevice()
|
| D | op_level_cost_estimator.cc | 1175 OpInfo::TensorProperties* a_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum()
1179 OpInfo::TensorProperties* b_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum()
1327 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations()
1335 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations()
|
| D | virtual_scheduler.cc | 764 *op_info.add_inputs() = input; in CreateOpContext()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | special_selector.cc | 127 auto add_inputs = graph.FindInputs(add_node->id); in TryFCFCAdd() local
128 if (add_inputs.size() != 2) { in TryFCFCAdd()
131 auto fc1_output_id = add_inputs[0]->id + add_inputs[1]->id - fc0_output_id; in TryFCFCAdd()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_test.py | 211 input_tensor, scale = custom.add_inputs(input_tensor, scale)
242 a, x, b = custom.add_inputs(a, x, b)
270 x, = custom.add_inputs(x)
310 b, = custom.add_inputs(b)
|
| D | op_hint.py | 439 def add_inputs(self, *args, **kwargs): member in OpHint
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.lite.-op-hint.pbtxt | 66 name: "add_inputs"
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.cc | 174 TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; in AddOpHelper() local
175 TF_AddInputList(desc, add_inputs, 2); in AddOpHelper()
201 TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; in AddWithCtrlDependency() local
202 TF_AddInputList(desc, add_inputs, 2); in AddWithCtrlDependency()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.cc | 113 *op_context.op_info.add_inputs() = GetTensorProperties(tensor); in Predict()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | op_callbacks_test.py | 485 add_inputs = instrument.graph_inputs[instrument.graph_op_types.index(
487 self.assertIsInstance(add_inputs, tuple)
488 self.assertEqual(len(add_inputs), 2)
490 compat.as_bytes(add_inputs[0].op.op_def.name), _CONSTANT_OP)
491 self.assertEqual(compat.as_bytes(add_inputs[1].op.op_def.name), _UNIQUE_OP)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 42 op_context.op_info.add_inputs()->Swap(&input); in PredictExecutionTime()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 1023 input = op->add_inputs(); in TEST()
1025 input = op->add_inputs(); in TEST()
1031 input = op->add_inputs(); in TEST()
1033 input = op->add_inputs(); in TEST()
1039 input = op->add_inputs(); in TEST()
1041 input = op->add_inputs(); in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model.cc | 1514 ModelProto::Node* input_proto = node_proto->add_inputs(); in ToProto()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1674 sig_def_builder.add_inputs(inputs_buf); in CreateSignatureDefs()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_generated.h | 10756 void add_inputs(flatbuffers::Offset<flatbuffers::Vector<int32_t>> inputs) {
10809 builder_.add_inputs(inputs);
10909 void add_inputs(flatbuffers::Offset<flatbuffers::Vector<int32_t>> inputs) {
10944 builder_.add_inputs(inputs);
11242 …void add_inputs(flatbuffers::Offset<flatbuffers::Vector<flatbuffers::Offset<tflite::TensorMap>>> i…
11276 builder_.add_inputs(inputs);
|