Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc27 tensorflow::gtl::ArraySlice<int64> input_dimensions, in ValidatePaddingValues()
43 tensorflow::gtl::ArraySlice<int64> input_dimensions, in MakePadding()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h449 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
459 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
Dcuda_dnn.cc3695 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
3956 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3991 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4026 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4060 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4098 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4136 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4264 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoDepthConcatenate()
4322 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1187 const dnn::BatchDescriptor &input_dimensions, in ThenMatMul()
1207 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
1229 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
1268 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1292 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1313 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1334 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
1361 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
1385 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
1491 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in ThenDepthConcatenate()
[all …]