Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h450 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
460 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
Dcuda_dnn.cc3696 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
3958 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
3993 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4028 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4062 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4100 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4138 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4281 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
4324 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc257 std::vector<int64> output_dimensions(window.dimensions_size()); in InferWindowOutputShape() local
764 std::vector<int64> output_dimensions(ShapeUtil::Rank(lhs)); in InferDegenerateDimensionBroadcastShape() local
Duser_computation.cc3429 std::vector<int64> output_dimensions; in Visit() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1188 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul()
1208 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
1230 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
1270 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
1294 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
1315 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
1336 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
1363 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
1387 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
1569 const dnn::BatchDescriptor &output_dimensions, in ThenReshape()
[all …]