/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | while_op.cc | 250 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 D | case_op.cc | 275 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 D | if_op.cc | 296 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 D | xla_launch_util.cc | 258 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 D | einsum_spmd_expander.cc | 404 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 D | compile_mlir_util.cc | 307 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 D | tf_xla_mlir_translate.cc | 98 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 D | ops_testutil.h | 99 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 D | restore_v2_op_test.cc | 44 std::function<T(int)> input_mapping) { in MakeInput() argument 46 test::FillFn(&input, input_mapping); in MakeInput()
|
H A D | restore_op_test.cc | 58 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 D | graph_executor.cpp | 1032 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 D | tpu_dynamic_layout_pass.cc | 223 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 D | mod.rs | 381 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 D | xla_compiler.cc | 180 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 D | xla_helpers.h | 135 std::vector<int> input_mapping; member
|
H A D | xla_compiler_test.cc | 539 EXPECT_THAT(result.input_mapping, ::testing::ElementsAre(0, 1)); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
H A D | dtensor_device.cc | 1663 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 D | legalize-xla-framework.mlir | 5 func.func @buffer_type(%arg: !xla_framework.buffer {xla_framework.input_mapping = 0 : i64})
|
/aosp_15_r20/external/harfbuzz_ng/src/ |
H A D | hb-ot-layout-gsubgpos.hh | 2059 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 D | vmlinux.h | 30085 …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 D | vmlinux.h | 33662 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
|
H A D | vmlinux_600.h | 33662 …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 D | vmlinux.h | 61198 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
|
H A D | vmlinux_602.h | 61198 …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 D | vmlinux_608.h | 56499 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
|