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.h400 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
410 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
Dcuda_dnn.cc5268 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
5473 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
5498 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
5523 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
5548 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
5573 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
5603 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
5633 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
5747 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
5791 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h390 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
400 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
Drocm_dnn.cc3704 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
3896 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
3904 const dnn::BatchDescriptor& output_dimensions, in IsSame()
3919 const dnn::BatchDescriptor& output_dimensions, in find()
3937 const dnn::BatchDescriptor& output_dimensions, in insert()
3990 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4056 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4087 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackwardImpl()
4198 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4211 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.h1549 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
1560 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
1571 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
1583 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
1596 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
1609 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
1751 const dnn::BatchDescriptor& output_dimensions, in DoReshape()
1863 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperateScaledQuantized()
Dstream.cc546 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul()
564 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
584 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
620 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
642 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
662 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
683 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
703 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
730 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
755 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc172 std::vector<int64> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
739 std::vector<int64> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6624 nvinfer1::Dims output_dimensions; in ConvertResize() local