| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | padding.cc | 26 Status ValidatePaddingValues(absl::Span<const int64> input_dimensions, in ValidatePaddingValues()
42 absl::Span<const int64> input_dimensions, in MakePadding()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 399 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
409 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
|
| D | cuda_dnn.cc | 5267 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
5471 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
5496 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
5521 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
5546 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
5571 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
5601 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
5631 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
5730 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoDepthConcatenate()
5789 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoElementwiseOperate()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLWinogradOutputTransformKernel.cpp | 63 const Size2D input_dimensions = winograd_info.input_dimensions; in validate_arguments() local
150 const Size2D input_dimensions = winograd_info.input_dimensions; in configure() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 389 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
399 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
|
| D | rocm_dnn.cc | 3703 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
3894 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3903 const dnn::BatchDescriptor& input_dimensions, in IsSame()
3918 const void* p, const dnn::BatchDescriptor& input_dimensions, in find()
3936 const void* p, const dnn::BatchDescriptor& input_dimensions, in insert()
3988 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4054 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4085 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackwardImpl()
4196 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4209 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1547 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
1558 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
1569 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
1581 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
1594 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
1607 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
1713 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoSpaceConcatenate()
1749 const dnn::BatchDescriptor& input_dimensions, in DoReshape()
1777 const dnn::BatchDescriptor& input_dimensions, in DoDepthToSpace()
1806 const dnn::BatchDescriptor& input_dimensions, in DoSpaceToDepth()
[all …]
|
| D | stream.cc | 545 const dnn::BatchDescriptor &input_dimensions, in ThenMatMul()
563 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
583 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
618 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
640 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
660 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
681 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
701 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
728 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
753 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | quantize_and_dequantize_op.cc | 167 absl::Span<const int64> input_dimensions = input_shape.dimensions(); in Compile() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Winograd.cpp | 443 const Size2D input_dimensions = winograd_info.input_dimensions; in winograd_output_transform() local
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 707 const Size2D input_dimensions = winograd_info.input_dimensions; in compute_winograd_output_transform_shape() local
|