Home
last modified time | relevance | path

Searched defs:output_dimensions (Results 1 – 8 of 8) 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.cc3529 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/
Drocm_dnn.h391 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
401 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
Drocm_dnn.cc3289 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/
Dperiodic_resample_op.h36 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/
Dsparse_to_dense.cc46 const int output_dimensions = NumElements(output_shape); in Resize() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1142 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/
Dshape_inference.cc169 std::vector<int64> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
720 std::vector<int64> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local