Searched defs:input_coords (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NESpaceToDepthLayerKernel.cpp | 129 Coordinates input_coords{ in_x, in_y, z, batch_id }; in run() local
148 Coordinates input_coords{ z, in_x, in_y, batch_id }; in run() local
|
| H A D | NESpaceToBatchLayerKernel.cpp | 193 Coordinates input_coords{ in_x, in_y, z, w }; in run() local
219 Coordinates input_coords{ z, in_x, in_y, w }; in run() local
|
| H A D | NETileKernel.cpp | 113 …Coordinates input_coords{ x % src_shape[0], y % src_shape[1], z % src_shape[2], w % src_shape[3] … in run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_util.cc | 208 std::vector<InputCoords> input_coords; in GetComputationInputs() local
258 const std::vector<InputCoords>& input_coords, in GetInputTupleAllocations()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_execute_op.cc | 90 xla::Backend* backend, const std::vector<InputCoords>& input_coords, in GetInputTuples()
402 const std::vector<InputCoords>& input_coords, bool release_inputs, in ExecuteComputation()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Unstack.cpp | 78 … const Coordinates input_coords = expand_coordinates(id, axis, slice, unpacked_num_dimensions); in get_slice() local
|