Home
last modified time | relevance | path

Searched refs:input_mapping (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc227 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()
Dif_op.cc200 int num_inputs = then_result.input_mapping.size(); in Compile()
203 int input_num = then_result.input_mapping[i] + 1; in Compile()
Dcase_op.cc212 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/
Dops_testutil.h116 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()
Drestore_v2_op_test.cc41 std::function<T(int)> input_mapping) { in MakeInput() argument
43 test::FillFn(&input, input_mapping); in MakeInput()
Drestore_op_test.cc55 std::function<T(int)> input_mapping) { in MakeInput() argument
57 test::FillFn(&input, input_mapping); in MakeInput()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h251 std::vector<int> input_mapping; member
Dxla_compiler.cc1092 &arg_expressions, &result->input_mapping, &result->xla_input_shapes, in CompileGraph()
Dxla_compiler_test.cc418 EXPECT_THAT(result.input_mapping, ::testing::ElementsAre(0, 1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc221 int arg_num = kernel->input_mapping[i]; in PopulateInputs()