| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 58 auto input = op_info->add_inputs(); in DescribeMatrix()
90 auto input = op_info->add_inputs(); in DescribeArbitraryRankInput()
167 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeConvolution()
168 DescribeTensor4D(kx, ky, iz2, oz, op_context.op_info.add_inputs()); in DescribeConvolution()
184 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeDepthwiseConv2dNative()
185 DescribeTensor4D(kx, ky, iz2, cm, op_context.op_info.add_inputs()); in DescribeDepthwiseConv2dNative()
211 DescribeTensor4D(batch, ix, iy, iz1, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
214 DescribeTensor4D(batch, iz1, ix, iy, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
217 DescribeTensor4D(kx, ky, iz2, oz, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
220 DescribeTensor4D(oz, iz2, kx, ky, op_context.op_info.add_inputs()); in DescribeFusedConv2DBiasActivation()
[all …]
|
| D | utils.cc | 270 *op_info.add_inputs() = input; in BuildOpInfoWithoutDevice()
|
| D | op_level_cost_estimator.cc | 841 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations()
849 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations()
|
| D | virtual_scheduler.cc | 639 *op_info.add_inputs() = input; in GetCurrNode()
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/ |
| D | evaluation_stage_test.cc | 43 config.add_inputs(kInputMapping); in GetIdentityStageConfig()
62 config.add_inputs(kInputMapping); in TEST()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_test.py | 236 input_tensor, scale = custom.add_inputs(input_tensor, scale)
264 a, x, b = custom.add_inputs(a, x, b)
289 x, = custom.add_inputs(x)
326 b, = custom.add_inputs(b)
|
| D | op_hint.py | 427 def add_inputs(self, *args, **kwargs): member in OpHint
|
| /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/tools/api/golden/v1/ |
| D | tensorflow.lite.-op-hint.pbtxt | 66 name: "add_inputs"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.lite.-op-hint.pbtxt | 66 name: "add_inputs"
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 40 op_context.op_info.add_inputs()->Swap(&input); in PredictExecutionTime()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 108 auto* input = operation->add_inputs(); in AddOperationToEnqueueRequest()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 686 auto* remote_op_input = remote_op->add_inputs(); in EagerRemoteExecute()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_generated.h | 8351 void add_inputs(flatbuffers::Offset<flatbuffers::Vector<int32_t>> inputs) {
8399 builder_.add_inputs(inputs);
8492 void add_inputs(flatbuffers::Offset<flatbuffers::Vector<int32_t>> inputs) {
8527 builder_.add_inputs(inputs);
|