Home
last modified time | relevance | path

Searched defs:output_dimensions (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h419 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
429 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
Dcuda_dnn.cc4062 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
4265 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4290 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4315 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4340 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4365 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4395 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4425 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4539 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
4582 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h443 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
453 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
Drocm_dnn.cc3815 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
4005 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4013 const dnn::BatchDescriptor& output_dimensions, in IsSame()
4028 const dnn::BatchDescriptor& output_dimensions, in find()
4046 const dnn::BatchDescriptor& output_dimensions, in insert()
4099 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4165 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4196 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackwardImpl()
4307 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4320 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense.cc42 const int output_dimensions = NumElements(output_shape); in Resize() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1648 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
1659 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
1670 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
1682 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
1695 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
1708 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
1850 const dnn::BatchDescriptor& output_dimensions, in DoReshape()
1962 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperateScaledQuantized()
Dstream.cc749 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul()
767 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
787 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
823 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
845 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
865 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
886 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
906 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
933 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
958 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc171 std::vector<int64> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
750 std::vector<int64> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6144 nvinfer1::Dims output_dimensions; in ConvertResize() local