Searched refs:input_mapping (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | while_op.cc | 227 const std::vector<int>& input_mapping, in BuildWhile() argument 237 padded_while_outputs[input_mapping[while_result_index]] = in BuildWhile() 465 int num_inputs = body.input_mapping.size(); in Compile() 468 int input_num = body.input_mapping[i]; in Compile() 525 BuildWhile(ctx, wrapped_cond, wrapped_body, init, body.input_mapping, in Compile()
|
D | if_op.cc | 200 int num_inputs = then_result.input_mapping.size(); in Compile() 203 int input_num = then_result.input_mapping[i] + 1; in Compile()
|
D | case_op.cc | 212 int num_inputs = branch_results[0].input_mapping.size(); in Compile() 215 int input_num = branch_results[0].input_mapping[i] + 1; 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/tf2xla/ |
D | xla_compiler.h | 251 std::vector<int> input_mapping; member
|
D | xla_compiler.cc | 1092 &arg_expressions, &result->input_mapping, &result->xla_input_shapes, in CompileGraph()
|
D | xla_compiler_test.cc | 418 EXPECT_THAT(result.input_mapping, ::testing::ElementsAre(0, 1)); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_launch_util.cc | 221 int arg_num = kernel->input_mapping[i]; in PopulateInputs()
|