Searched defs:input_mapping (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.h | 99 void AddInput(const TensorShape& shape, std::function<T(int)> input_mapping) { in AddInput()
|
| D | restore_v2_op_test.cc | 44 std::function<T(int)> input_mapping) { in MakeInput()
|
| D | restore_op_test.cc | 58 std::function<T(int)> input_mapping) { in MakeInput()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | einsum_spmd_expander.cc | 404 for (const auto& input_mapping : input_mappings) in MaybeRelayoutInputs() local 438 for (const auto& input_mapping : input_mappings) in MaybeRelayoutInputs() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.h | 135 std::vector<int> input_mapping; member
|
| D | xla_compiler.cc | 180 xla::Shape* output_shape, absl::Span<int const> input_mapping) { in BuildComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_dynamic_layout_pass.cc | 223 const auto& input_mapping = std::get<1>(execute_and_input_mapping); in HandleCompileAndExecutes() local
|
| /external/crosvm/devices/src/virtio/video/decoder/backend/ |
| D | mod.rs | 294 let input_mapping = MemoryMappingBuilder::new(input_shm.size() as usize) in decode_h264_generic() localVariable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 250 const std::vector<int>& input_mapping, in BuildWhile()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 324 absl::Span<const int> input_mapping, in GetOrCreateTensorForOutput()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 307 void GetInputMappingForMlir(int num_inputs, std::vector<int>* input_mapping) { in GetInputMappingForMlir()
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device.cc | 1663 auto input_mapping = function.input_index_map; in ExecuteRegularOperation() local
|
| /external/libbpf/.github/actions/build-selftests/ |
| D | vmlinux.h | 75192 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
|
| /external/bcc/libbpf-tools/x86/ |
| D | vmlinux_505.h | 98948 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
|
| D | vmlinux.h | 98948 …int (*input_mapping)(struct hid_device *, struct hid_input *, struct hid_field *, struct hid_usage… member
|