Searched refs:input_mapping (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 431 std::vector<int>* input_mapping, std::vector<xla::Shape>* input_shapes, in BuildArguments() argument
438 input_mapping->clear(); in BuildArguments()
439 input_mapping->reserve(args.size()); in BuildArguments()
465 input_mapping->push_back(i); in BuildArguments()
478 input_mapping->insert(input_mapping->end(), resources.begin(), in BuildArguments()
480 if (input_mapping->empty()) { in BuildArguments()
484 std::vector<xla::Shape> arg_shapes(input_mapping->size()); in BuildArguments()
485 for (std::vector<int>::size_type i = 0; i < input_mapping->size(); ++i) { in BuildArguments()
488 XLAShapeForArgument(args[(*input_mapping)[i]], &arg_shapes[i])); in BuildArguments()
518 std::vector<xla::ComputationDataHandle> arg_handles(input_mapping->size()); in BuildArguments()
[all …]
|
| D | xla_compiler.h | 206 std::vector<int> input_mapping; member
318 std::vector<int>* input_mapping,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 236 int num_inputs = body.input_mapping.size(); in Compile()
239 int input_num = body.input_mapping[i]; in Compile()
272 ctx->SetOutput(body.input_mapping[i], in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.h | 116 void AddInput(const TensorShape& shape, std::function<T(int)> input_mapping) { in AddInput() argument
117 test::FillFn(AddInput(DataTypeToEnum<T>::v(), shape), input_mapping); in AddInput()
|
| D | restore_v2_op_test.cc | 41 std::function<T(int)> input_mapping) { in MakeInput() argument
43 test::FillFn(&input, input_mapping); in MakeInput()
|
| D | restore_op_test.cc | 55 std::function<T(int)> input_mapping) { in MakeInput() argument
57 test::FillFn(&input, input_mapping); in MakeInput()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_launch_op.cc | 283 int arg_num = kernel->input_mapping[i]; in Compute()
|