Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h425 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
435 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
Dcuda_dnn.cc4071 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
4274 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4299 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4324 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4349 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
4374 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4404 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4434 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4548 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
4591 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h410 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
420 const dnn::BatchDescriptor& output_dimensions, in DoMatMulQuantized()
Drocm_dnn.cc3578 const dnn::BatchDescriptor& output_dimensions, in DoMatMul()
3768 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
3778 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
3807 const dnn::BatchDescriptor& output_dimensions, in DoPoolForward()
3837 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
3850 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
3946 const dnn::BatchDescriptor& output_dimensions, in DoPoolBackward()
4191 dnn::BatchDescriptor output_dimensions = in DoDepthConcatenate() local
4234 const dnn::BatchDescriptor& output_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconv.cc291 const BHWC& input_dimensions, const BHWC& output_dimensions, in GetUniformBufferForConvShared()
947 const auto& output_dimensions = buffers.find(output_id)->second; in Convolution() local
1004 const auto& output_dimensions = buffers.find(output_id)->second; in Convolution1x1() local
1062 const auto& output_dimensions = buffers.find(output_id)->second; in ConvolutionGeneric() local
1116 const auto& output_dimensions = buffers.find(output_id)->second; in ConvolutionPrecise() local
1174 const auto& output_dimensions = buffers.find(output_id)->second; in ConvolutionPrecise1x1PowerVR() local
Dresize.cc134 const auto& output_dimensions = buffers.find(output_id)->second; in Resize() local
Ddepthwise_conv.cc654 const auto& output_dimensions = buffers.find(output_id)->second; in DepthWiseConv3x3Stride1x1() local
718 const auto& output_dimensions = buffers.find(output_id)->second; in DepthWiseConv3x3Stride2() local
/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.cc1284 const dnn::BatchDescriptor &output_dimensions, in ThenMatMul()
1304 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
1326 const dnn::BatchDescriptor &output_dimensions, in ThenMatMulQuantized()
1366 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
1391 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
1413 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
1436 const dnn::BatchDescriptor &output_dimensions, in ThenPoolForward()
1458 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
1488 const dnn::BatchDescriptor &output_dimensions, in ThenPoolBackward()
1515 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
730 std::vector<int64> output_dimensions(lhs.rank()); in InferDegenerateDimensionBroadcastShape() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5509 nvinfer1::Dims output_dimensions; in ConvertResize() local