| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | padding.cc | 26 Status ValidatePaddingValues(absl::Span<const int64_t> input_dimensions, in ValidatePaddingValues()
42 absl::Span<const int64_t> input_dimensions, in MakePadding()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.h | 411 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
421 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
|
| H A D | cuda_dnn.cc | 5957 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
6224 const dnn::BatchDescriptor& input_dimensions, DeviceMemoryBase input_data, in DoPoolForward()
6290 const dnn::BatchDescriptor& input_dimensions, DeviceMemoryBase input_data, in DoPoolBackward()
6432 BatchDescriptorSlice input_dimensions, in DoDepthConcatenate()
|
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/ |
| H A D | ClWinogradOutputTransformKernel.cpp | 67 const Size2D input_dimensions = winograd_info.input_dimensions; in validate_arguments() local
150 const Size2D input_dimensions = winograd_info.input_dimensions; in configure() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.h | 415 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
425 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
|
| H A D | rocm_dnn.cc | 3828 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
4021 const dnn::BatchDescriptor& input_dimensions, DeviceMemoryBase input_data, in DoPoolForward()
4090 const dnn::BatchDescriptor& input_dimensions, in IsSame()
4105 const void* p, const dnn::BatchDescriptor& input_dimensions, in find()
4123 const void* p, const dnn::BatchDescriptor& input_dimensions, in insert()
4176 const dnn::BatchDescriptor& input_dimensions, DeviceMemoryBase input_data, in DoPoolBackward()
4421 Stream* stream, absl::Span<const dnn::BatchDescriptor> input_dimensions, in DoDepthConcatenate()
4480 absl::Span<const dnn::BatchDescriptor> input_dimensions, in DoElementwiseOperate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.cc | 538 const dnn::BatchDescriptor &input_dimensions, in ThenMatMul()
556 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
576 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
673 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, // non-absl ok in ThenDepthConcatenate()
701 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, // non-absl ok in ThenSpaceConcatenate()
745 Stream &Stream::ThenReshape(const dnn::BatchDescriptor &input_dimensions, in ThenReshape()
762 const dnn::BatchDescriptor &input_dimensions, in ThenDepthToSpace()
781 const dnn::BatchDescriptor &input_dimensions, in ThenSpaceToDepth()
801 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, // non-absl ok in ThenElementwiseOperate()
822 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, // non-absl ok in ThenElementwiseOperateScaledQuantized()
|
| H A D | dnn.h | 1738 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, // non-absl ok in DoSpaceConcatenate()
1774 const dnn::BatchDescriptor& input_dimensions, in DoReshape()
1802 const dnn::BatchDescriptor& input_dimensions, in DoDepthToSpace()
1831 const dnn::BatchDescriptor& input_dimensions, in DoSpaceToDepth()
1887 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, // non-absl ok in DoElementwiseOperateScaledQuantized()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | fake_quantize_ops.cc | 265 absl::Span<const int64_t> input_dimensions = input_shape.dimensions(); in Compile() local
320 absl::Span<const int64_t> input_dimensions = input_shape.dimensions(); in Compile() local
|
| H A D | quantize_and_dequantize_op.cc | 167 absl::Span<const int64_t> input_dimensions = input_shape.dimensions(); in Compile() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Winograd.cpp | 443 const Size2D input_dimensions = winograd_info.input_dimensions; in winograd_output_transform() local
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/ |
| H A D | ShapeCalculator.h | 681 const Size2D input_dimensions = winograd_info.input_dimensions; in compute_winograd_output_transform_shape() local
|