Searched defs:output_dimensions (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 400 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() 410 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
|
D | cuda_dnn.cc | 3529 const dnn::BatchDescriptor& output_dimensions, in DoMatMul() 3732 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() 3757 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() 3782 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() 3807 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() 3832 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() 3862 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() 3892 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() 4006 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local 4049 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 391 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized() 401 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
|
D | rocm_dnn.cc | 3289 const dnn::BatchDescriptor& output_dimensions, in DoMatMul() 3479 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() 3489 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() 3518 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward() 3548 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() 3561 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() 3657 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward() 3902 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local 3945 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
|
/external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
D | periodic_resample_op.h | 36 InputIndexer(const std::vector<tensorflow::int64>& output_dimensions, in InputIndexer() 181 const std::vector<tensorflow::int64>& output_dimensions, in ComputeDimensionCeiling()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | sparse_to_dense.cc | 46 const int output_dimensions = NumElements(output_shape); in Resize() local
|
/external/tensorflow/tensorflow/stream_executor/ |
D | stream.cc | 1142 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul() 1162 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized() 1184 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized() 1224 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward() 1249 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward() 1271 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward() 1294 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward() 1316 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward() 1346 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward() 1373 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | shape_inference.cc | 169 std::vector<int64> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local 720 std::vector<int64> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local
|