Home
last modified time | relevance | path

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

/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
438 for (const auto& input_mapping : input_mappings) in MaybeRelayoutInputs() local
/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()
H A Drestore_v2_op_test.cc44 std::function<T(int)> input_mapping) { in MakeInput()
H A Drestore_op_test.cc58 std::function<T(int)> input_mapping) { in MakeInput()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_helpers.h135 std::vector<int> input_mapping; member
H A Dxla_compiler.cc180 xla::Shape* output_shape, absl::Span<int const> input_mapping) { in BuildComputation()
/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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dwhile_op.cc250 const std::vector<int>& input_mapping, in BuildWhile()
/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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.cc324 absl::Span<const int> input_mapping, in GetOrCreateTensorForOutput()
/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()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dgraph_executor.cpp1032 std::unordered_map<Value*, Value*> input_mapping; in replaceBlockWithFallbackGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc1663 auto input_mapping = function.input_index_map; in ExecuteRegularOperation() local
/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/system/bpf/include/vmlinux/android15_6.6/arm64/
H A Dvmlinux.h98260 …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
H A Dvmlinux.h56499 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
/aosp_15_r20/external/bcc/libbpf-tools/riscv/
H A Dvmlinux_602.h113159 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
H A Dvmlinux.h113159 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
/aosp_15_r20/system/bpf/include/vmlinux/android15_6.6/x86_64/
H A Dvmlinux.h107956 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member