Home
last modified time | relevance | path

Searched refs:input_num (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddynamic_stitch_op.cc70 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 …]
Dwhile_op.cc468 int input_num = body.input_mapping[i]; in Compile() local
478 } else if (ctx->input_type(input_num) == DT_RESOURCE) { in Compile()
480 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
482 } else if (IsTensorListInput(ctx, input_num)) { in Compile()
487 OP_REQUIRES_OK(ctx, GetTensorListBufferShape(ctx->Input(input_num), in Compile()
497 OP_REQUIRES_OK(ctx, InitializeTensorList(ctx->Input(input_num), in Compile()
500 inputs[i] = ctx->Input(input_num); in Compile()
503 inputs[i] = ctx->Input(input_num); in Compile()
Dif_op.cc203 int input_num = then_result.input_mapping[i] + 1; in Compile() local
213 } else if (ctx->input_type(input_num) == DT_RESOURCE) { in Compile()
215 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
Dcase_op.cc215 int input_num = branch_results[0].input_mapping[i] + 1; in Compile() local
225 } else if (ctx->input_type(input_num) == DT_RESOURCE) { in Compile()
227 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
/external/tensorflow/tensorflow/core/kernels/
Ddynamic_stitch_op.cc104 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()
254 auto OnInputNumber = [&](int input_num) { in Compute() argument
[all …]
Dsummary_op.cc139 for (int input_num = 0; input_num < c->num_inputs(); input_num++) { in Compute() local
140 const Tensor& in = c->input(input_num); in Compute()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc449 for (int64 input_num = 0; input_num < ctx->num_inputs(); ++input_num) { in BuildXlaCompilerArguments() local
450 XlaCompiler::Argument& arg = (*args)[input_num]; in BuildXlaCompilerArguments()
451 if (constant_args.count(input_num) > 0) { in BuildXlaCompilerArguments()
453 const Tensor& input = constant_args.at(input_num); in BuildXlaCompilerArguments()
459 } else if (variable_args.count(input_num) == 0) { in BuildXlaCompilerArguments()
461 const Tensor& input = ctx->input(input_num); in BuildXlaCompilerArguments()
473 const Tensor& input = ctx->input(input_num); in BuildXlaCompilerArguments()
475 const OptionalTensor& variable = variable_args.at(input_num); in BuildXlaCompilerArguments()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.cc269 int64 input_num = xla_args.size(); in ConvertGraphToXla() local
284 /*param_number=*/input_num, /*param_index=*/{}}); in ConvertGraphToXla()
285 ++input_num; in ConvertGraphToXla()