Home
last modified time | relevance | path

Searched defs:input_dimensions (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc26 Status ValidatePaddingValues(absl::Span<const int64> input_dimensions, in ValidatePaddingValues()
42 absl::Span<const int64> input_dimensions, in MakePadding()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h424 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
434 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
Dcuda_dnn.cc4070 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
4272 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4297 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4322 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4347 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
4372 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4402 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4432 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4531 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoDepthConcatenate()
4589 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h409 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
419 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized()
Drocm_dnn.cc3577 const dnn::BatchDescriptor& input_dimensions, in DoMatMul()
3766 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3776 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3805 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward()
3835 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
3848 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
3944 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward()
4174 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoDepthConcatenate()
4232 port::ArraySlice<dnn::BatchDescriptor> input_dimensions, in DoElementwiseOperate()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconv.cc245 BHWC input_dimensions{params.weights.shape.o, params.weights.shape.h, in ReorderWeightsForConvShared() local
291 const BHWC& input_dimensions, const BHWC& output_dimensions, in GetUniformBufferForConvShared()
946 const auto& input_dimensions = buffers.find(input_id)->second; in Convolution() local
1003 const auto& input_dimensions = buffers.find(input_id)->second; in Convolution1x1() local
1061 const auto& input_dimensions = buffers.find(input_id)->second; in ConvolutionGeneric() local
1115 const auto& input_dimensions = buffers.find(input_id)->second; in ConvolutionPrecise() local
1173 const auto& input_dimensions = buffers.find(input_id)->second; in ConvolutionPrecise1x1PowerVR() local
Dresize.cc133 const auto& input_dimensions = buffers.find(input_id)->second; in Resize() local
Ddepthwise_conv.cc653 const auto& input_dimensions = buffers.find(input_id)->second; in DepthWiseConv3x3Stride1x1() local
717 const auto& input_dimensions = buffers.find(input_id)->second; in DepthWiseConv3x3Stride2() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dquantize_and_dequantize_op.cc167 absl::Span<const int64> input_dimensions = input_shape.dimensions(); in Compile() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1283 const dnn::BatchDescriptor &input_dimensions, in ThenMatMul()
1303 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
1325 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized()
1364 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1389 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1411 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1434 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward()
1456 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
1486 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
1513 const dnn::BatchDescriptor &input_dimensions, in ThenPoolBackward()
[all …]