Home
last modified time | relevance | path

Searched defs:input_dimensions (Results 1 – 7 of 7) 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.h399 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
409 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
Dcuda_dnn.cc3528 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
3730 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3755 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3780 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3805 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3830 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
3860 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
3890 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
3989 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoDepthConcatenate()
4047 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h390 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
400 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
Drocm_dnn.cc3288 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
3477 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3487 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3516 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3546 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
3559 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
3655 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
3885 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoDepthConcatenate()
3943 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/contrib/periodic_resample/kernels/
Dperiodic_resample_op.h45 auto input_dimensions = TensorShapeToVector(input_shape); in InputIndexer() local
182 const std::vector<tensorflow::int64>& input_dimensions) { in ComputeDimensionCeiling()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1141 const dnn::BatchDescriptor &input_dimensions, in ThenMatMul()
1161 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
1183 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
1222 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1247 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1269 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1292 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1314 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
1344 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
1371 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
[all …]