Searched refs:arg_handles (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 1052 std::vector<xla::XlaOp> arg_handles(input_to_args->size()); in BuildArguments() local
1091 arg_handles[i] = xla::GetTupleElement(tuple, i); in BuildArguments()
1104 arg_handles[i] = in BuildArguments()
1108 arg_handles[i] = xla::Parameter(builder, i, (*input_shapes)[i], in BuildArguments()
1120 arg_handles[i] = xla::SetDimensionSize( in BuildArguments()
1121 arg_handles[i], arg_handles[dynamic_size_param_index], in BuildArguments()
1148 arg_handles[i], builder)); in BuildArguments()
1159 xla::Reshape(arg_handles[i], arg.DimensionSizes()), arg.type); in BuildArguments()
1161 arg_expression = XlaExpression::XlaOp(arg_handles[i], arg.type); in BuildArguments()
1169 arg_expression = XlaExpression::TensorList(arg_handles[i]); in BuildArguments()
[all …]
|