| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_stitch_op.cc | 70 for (int input_num = 0; input_num < indices_input.size(); input_num++) { in Compile() local
73 XLAShapeToTensorShape(indices_input[input_num].shape(), in Compile()
75 const TensorShape& data_shape = data_shapes[input_num]; in Compile()
78 errors::InvalidArgument("data[", input_num, in Compile()
80 " does not start with indices[", input_num, in Compile()
84 input_num == 0 || SameExtraShape(data0_shape, indices0_shape, in Compile()
88 input_num, "].shape[", indices_shape.dims(), in Compile()
90 input_num, "].shape = ", data_shape.DebugString(), in Compile()
92 ", indices[", input_num, in Compile()
96 XlaHelpers::ReshapeLiteral(indices_input[input_num], in Compile()
[all …]
|
| D | while_op.cc | 488 int input_num = body.input_mapping[i]; in Compile() local
498 } else if (ctx->input_type(input_num) == DT_RESOURCE) { in Compile()
500 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
502 } else if (IsTensorListInput(ctx, input_num)) { in Compile()
503 xla::XlaOp input = ctx->Input(input_num); in Compile()
549 inputs[i] = ctx->Input(input_num); in Compile()
552 inputs[i] = ctx->Input(input_num); in Compile()
|
| D | case_op.cc | 287 int input_num = branch_results[0].input_mapping[i] + 1; in Compile() local
297 } else if (ctx->input_type(input_num) == DT_RESOURCE) { in Compile()
299 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
302 inputs[i] = ctx->Input(input_num); in Compile()
|
| D | if_op.cc | 275 int input_num = then_result.input_mapping[i] + 1; in Compile() local
285 } else if (ctx->input_type(input_num) == DT_RESOURCE) { in Compile()
287 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
290 inputs[i] = ctx->Input(input_num); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op.cc | 104 for (int input_num = 0; input_num < indices_inputs->size(); input_num++) { in CheckArgsAndAllocateResult() local
105 const Tensor& indices = (*indices_inputs)[input_num]; in CheckArgsAndAllocateResult()
106 const Tensor& data = (*data_inputs)[input_num]; in CheckArgsAndAllocateResult()
109 errors::InvalidArgument("data[", input_num, in CheckArgsAndAllocateResult()
111 " does not start with indices[", input_num, in CheckArgsAndAllocateResult()
114 c, input_num == 0 || SameExtraShape(data0, indices0, data, indices), in CheckArgsAndAllocateResult()
116 "Need data[0].shape[", indices0.dims(), ":] = data[", input_num, in CheckArgsAndAllocateResult()
119 ", data[", input_num, "].shape = ", data.shape().DebugString(), in CheckArgsAndAllocateResult()
121 ", indices[", input_num, in CheckArgsAndAllocateResult()
255 auto OnInputNumber = [&](int input_num) { in Compute() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/ |
| D | fuse_auto_input.cc | 61 int input_num = -1; in ApplyToNode() local
63 input_num++; in ApplyToNode()
65 input_values.push_back({input_id, input_num}); in ApplyToNode()
96 nodes_to_fuse.push_back({input_producer, input_num}); in ApplyToNode()
138 int extra_input_num = input_num; in ApplyToNode()
139 input_num = 0; in ApplyToNode()
189 replacements.push_back(MakeDataReplacement(i, input_num)); in ApplyToNode()
195 input_num, "[gid.x, gid.y, gid.z]$;\n"); in ApplyToNode()
201 input_num++; in ApplyToNode()
232 " = $input_data_", input_num, in ApplyToNode()
[all …]
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | merge_summary_op.cc | 54 for (int input_num = 0; input_num < TF_NumInputs(ctx); ++input_num) { in MergeSummaryOp_Compute() local
56 TF_GetInput(ctx, input_num, &input, status.get()); in MergeSummaryOp_Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 610 for (int64 input_num = 0; input_num < inputs.size(); ++input_num) { in BuildXlaCompilerArguments() local
611 const Tensor* input = inputs[input_num]; in BuildXlaCompilerArguments()
613 XlaCompiler::Argument& arg = out[input_num]; in BuildXlaCompilerArguments()
614 if (variable_info_lookup.count(input_num)) { in BuildXlaCompilerArguments()
617 const VariableInfo& variable = *variable_info_lookup[input_num]; in BuildXlaCompilerArguments()
636 if (absl::c_binary_search(must_be_constant_idxs, input_num)) { in BuildXlaCompilerArguments()
649 } else if (absl::c_binary_search(must_be_constant_idxs, input_num)) { in BuildXlaCompilerArguments()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 650 for (auto input_num : op.inputs) { in ConvertOp() local
651 if (input_num == -1) { in ConvertOp()
655 op_state.addOperands({vals_map.at(input_num)}); in ConvertOp()
1063 for (auto input_num : op->inputs) { in ConvertSubgraph() local
1067 if (input_num == -1) { in ConvertSubgraph()
1075 } else if (!vals_map.at(input_num)) { in ConvertSubgraph()
1076 auto& const_tensor = *subgraph.tensors[input_num]; in ConvertSubgraph()
1088 vals_map[input_num] = op_or_err.ValueOrDie()->getResult(0); in ConvertSubgraph()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2266 int64 input_num = in BuildDynamicShapeNodes() local
2269 TF_RET_CHECK(0 <= input_num && in BuildDynamicShapeNodes()
2270 input_num < replicate_input_edges.size()); in BuildDynamicShapeNodes()
2271 src = replicate_input_edges[input_num]->src(); in BuildDynamicShapeNodes()
2272 src_output = replicate_input_edges[input_num]->src_output(); in BuildDynamicShapeNodes()
3121 int64 input_num = params_info.IsPerReplicaArg(orig_arg_num) in BuildExecuteNodes() local
3129 const Edge* edge = replicate_input_edges[input_num]; in BuildExecuteNodes()
3130 VLOG(2) << "replicate_input_edges[" << input_num << "]"; in BuildExecuteNodes()
3155 << replicate_input_fan_in_nodes[input_num][core]->name() in BuildExecuteNodes()
3157 graph->AddEdge(replicate_input_fan_in_nodes[input_num][core], 0, in BuildExecuteNodes()
[all …]
|
| /external/cpuinfo/test/dmesg/ |
| D | padcod-10.1.log | 530 <5>[ 8.680127] [VFE]v4l2 subdev register input_num = 0
536 <5>[ 8.680266] [VFE]vfe sensor detect start! input_num = 0
958 <5>[ 9.059254] [VFE]v4l2 subdev register input_num = 1
959 <5>[ 9.059261] [VFE]dev->ccm_cfg[input_num]->ccm = gc0308,regster_ccm = gc0308.
|