Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dwhile_op.cc250 const std::vector<int>& input_mapping, in BuildWhile() argument
260 padded_while_outputs[input_mapping[while_result_index]] = in BuildWhile()
497 int num_inputs = body.input_mapping.size(); in Compile()
500 int input_num = body.input_mapping[i]; in Compile()
587 BuildWhile(ctx, wrapped_cond, wrapped_body, init, body.input_mapping, in Compile()
H A Dcase_op.cc275 int num_inputs = branch_results[0].input_mapping.size(); in Compile()
278 int input_num = branch_results[0].input_mapping[i] + 1; in Compile()
H A Dif_op.cc296 int num_inputs = then_result.input_mapping.size(); in Compile()
299 int input_num = then_result.input_mapping[i] + 1; in Compile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.cc258 int arg_num = compilation_result->input_mapping[i]; in PopulateInputs()
324 absl::Span<const int> input_mapping, in GetOrCreateTensorForOutput() argument
337 input_mapping[alias->parameter_number] - missing_ctx_input_prefix; in GetOrCreateTensorForOutput()
560 input_output_alias, compilation_result->input_mapping, in PopulateOutputs()
599 compilation_result->input_mapping, in PopulateOutputs()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Deinsum_spmd_expander.cc404 for (const auto& input_mapping : input_mappings) in MaybeRelayoutInputs() local
405 for (const auto& char_and_positions : input_mapping) in MaybeRelayoutInputs()
438 for (const auto& input_mapping : input_mappings) in MaybeRelayoutInputs() local
439 for (const auto& char_and_positions : input_mapping) in MaybeRelayoutInputs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dcompile_mlir_util.cc307 void GetInputMappingForMlir(int num_inputs, std::vector<int>* input_mapping) { in GetInputMappingForMlir() argument
308 input_mapping->resize(num_inputs, 0); in GetInputMappingForMlir()
309 std::iota(input_mapping->begin(), input_mapping->end(), 0); in GetInputMappingForMlir()
621 GetInputMappingForMlir(arg_shapes.size(), &compilation_result->input_mapping); in PopulateResultIOInfo()
837 compilation_result->input_mapping = remaining_params; in CompileGraphToXlaHlo()
H A Dtf_xla_mlir_translate.cc98 if (!compilation_result.input_mapping.empty()) in PrintHloModuleText()
100 << absl::StrJoin(compilation_result.input_mapping, ", ") << "}\n"; in PrintHloModuleText()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dops_testutil.h99 void AddInput(const TensorShape& shape, std::function<T(int)> input_mapping) { in AddInput() argument
100 test::FillFn(AddInput(DataTypeToEnum<T>::v(), shape), input_mapping); in AddInput()
H A Drestore_v2_op_test.cc44 std::function<T(int)> input_mapping) { in MakeInput() argument
46 test::FillFn(&input, input_mapping); in MakeInput()
H A Drestore_op_test.cc58 std::function<T(int)> input_mapping) { in MakeInput() argument
60 test::FillFn(&input, input_mapping); in MakeInput()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dgraph_executor.cpp1032 std::unordered_map<Value*, Value*> input_mapping; in replaceBlockWithFallbackGraph() local
1033 auto value_map = [&input_mapping](Value* v) { return input_mapping[v]; }; in replaceBlockWithFallbackGraph()
1035 input_mapping[inp] = graph->block()->addInput(); in replaceBlockWithFallbackGraph()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_dynamic_layout_pass.cc223 const auto& input_mapping = std::get<1>(execute_and_input_mapping); in HandleCompileAndExecutes() local
252 metadata.mutable_args(input_mapping[execute_arg_index]) in HandleCompileAndExecutes()
/aosp_15_r20/external/crosvm/devices/src/virtio/video/decoder/backend/
H A Dmod.rs381 let input_mapping = MemoryMappingBuilder::new(input_shm.size() as usize) in decode_h264_generic() localVariable
420 input_mapping in decode_h264_generic()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler.cc180 xla::Shape* output_shape, absl::Span<int const> input_mapping) { in BuildComputation() argument
282 for (int xla_arg = 0; xla_arg < input_mapping.size(); xla_arg++) { in BuildComputation()
283 argument_to_xla_arg[input_mapping[xla_arg]] = xla_arg; in BuildComputation()
1364 arg_shardings, &arg_expressions, &result->input_mapping, in CompileGraph()
1430 result->input_mapping)); in CompileGraph()
H A Dxla_helpers.h135 std::vector<int> input_mapping; member
H A Dxla_compiler_test.cc539 EXPECT_THAT(result.input_mapping, ::testing::ElementsAre(0, 1)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc1663 auto input_mapping = function.input_index_map; in ExecuteRegularOperation() local
1670 std::sort(input_mapping.begin(), input_mapping.end()); in ExecuteRegularOperation()
1672 for (const int global_index : input_mapping) { in ExecuteRegularOperation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
H A Dlegalize-xla-framework.mlir5 func.func @buffer_type(%arg: !xla_framework.buffer {xla_framework.input_mapping = 0 : i64})
/aosp_15_r20/external/harfbuzz_ng/src/
H A Dhb-ot-layout-gsubgpos.hh2059 const hb_map_t *input_mapping, /* old->new glyphid or class mapping */ in serialize()
2071 d = input_mapping->get (org); in serialize()
/aosp_15_r20/external/libbpf/.github/actions/build-selftests/
H A Dvmlinux.h30085 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
/aosp_15_r20/external/bcc/libbpf-tools/powerpc/
H A Dvmlinux.h33662 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
H A Dvmlinux_600.h33662 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
/aosp_15_r20/external/bcc/libbpf-tools/loongarch/
H A Dvmlinux.h61198 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
H A Dvmlinux_602.h61198 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
/aosp_15_r20/external/bcc/libbpf-tools/arm64/
H A Dvmlinux_608.h56499 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member

12