Home
last modified time | relevance | path

Searched defs:input_mapping (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h116 void AddInput(const TensorShape& shape, std::function<T(int)> input_mapping) { in AddInput()
Drestore_v2_op_test.cc41 std::function<T(int)> input_mapping) { in MakeInput()
Drestore_op_test.cc55 std::function<T(int)> input_mapping) { in MakeInput()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h251 std::vector<int> input_mapping; member
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc227 const std::vector<int>& input_mapping, in BuildWhile()