Home
last modified time | relevance | path

Searched defs:output_dimensions (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ops/
H A Dutils.cpp82 std::vector<int64_t> output_dimensions; in BuildSqueezedDimensions() local
96 std::vector<int64_t> output_dimensions( in BuildUnsqueezedDimensions() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.h412 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
422 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
H A Dcuda_dnn.cc5958 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
6225 const dnn::BatchDescriptor& output_dimensions, DeviceMemoryBase output_data, in DoPoolForward()
6291 const dnn::BatchDescriptor& output_dimensions, DeviceMemoryBase output_data, in DoPoolBackward()
6449 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.h416 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
426 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
H A Drocm_dnn.cc3829 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
4022 const dnn::BatchDescriptor& output_dimensions, DeviceMemoryBase output_data, in DoPoolForward()
4091 const dnn::BatchDescriptor& output_dimensions, in IsSame()
4106 const dnn::BatchDescriptor& output_dimensions, in find()
4124 const dnn::BatchDescriptor& output_dimensions, in insert()
4177 const dnn::BatchDescriptor& output_dimensions, DeviceMemoryBase output_data, in DoPoolBackward()
4438 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
4482 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.cc539 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul()
557 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
577 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
747 const dnn::BatchDescriptor &output_dimensions, in ThenReshape()
803 const dnn::BatchDescriptor &output_dimensions, in ThenElementwiseOperate()
824 const dnn::BatchDescriptor &output_dimensions, in ThenElementwiseOperateScaledQuantized()
H A Ddnn.h1776 const dnn::BatchDescriptor& output_dimensions, in DoReshape()
1889 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperateScaledQuantized()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsparse_to_dense.cc42 const int output_dimensions = NumElements(output_shape); in Resize() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag_unpack.cpp195 std::vector<int64_t> output_dimensions = { in _qembeddingbag_nbit_unpack_helper() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/
H A DEmbeddingBag.cu553 std::vector<int64_t> output_dimensions = { in qembeddingbag_4bit_unpack() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc171 std::vector<int64_t> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
754 std::vector<int64_t> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference.cc1377 std::vector<int64_t> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local