Searched defs:input_mapping (Results 1 – 9 of 9) sorted by relevance
99 void AddInput(const TensorShape& shape, std::function<T(int)> input_mapping) { in AddInput()
44 std::function<T(int)> input_mapping) { in MakeInput()
58 std::function<T(int)> input_mapping) { in MakeInput()
141 std::vector<int> input_mapping; member
178 xla::Shape* output_shape, absl::Span<int const> input_mapping) { in BuildComputation()
240 const auto& input_mapping = std::get<1>(execute_and_input_mapping); in HandleCompileAndExecutes() local
245 const std::vector<int>& input_mapping, in BuildWhile()
304 absl::Span<const int> input_mapping, in GetOrCreateTensorForOutput()
209 void GetInputMappingForMlir(int num_inputs, std::vector<int>* input_mapping) { in GetInputMappingForMlir()