Home
last modified time | relevance | path

Searched defs:input_dimensions (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc26 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/
Dcuda_dnn.h418 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
428 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
Dcuda_dnn.cc4061 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
4263 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4288 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4313 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4338 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4363 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4393 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4423 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4522 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoDepthConcatenate()
4580 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h442 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
452 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
Drocm_dnn.cc3814 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
4003 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4012 const dnn::BatchDescriptor& input_dimensions, in IsSame()
4027 const void* p, const dnn::BatchDescriptor& input_dimensions, in find()
4045 const void* p, const dnn::BatchDescriptor& input_dimensions, in insert()
4097 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4163 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4194 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackwardImpl()
4305 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4318 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dquantize_and_dequantize_op.cc167 absl::Span<const int64> input_dimensions = input_shape.dimensions(); in Compile() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1646 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
1657 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
1668 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
1680 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
1693 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
1706 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
1812 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoSpaceConcatenate()
1848 const dnn::BatchDescriptor& input_dimensions, in DoReshape()
1876 const dnn::BatchDescriptor& input_dimensions, in DoDepthToSpace()
1905 const dnn::BatchDescriptor& input_dimensions, in DoSpaceToDepth()
[all …]
Dstream.cc748 const dnn::BatchDescriptor &input_dimensions, in ThenMatMul()
766 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
786 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
821 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
843 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
863 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
884 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
904 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
931 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
956 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
[all …]