Home
last modified time | relevance | path

Searched defs:output_descriptor (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h370 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
383 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
396 const dnn::BatchDescriptor& output_descriptor, in DoSeparableConvolve()
/external/ComputeLibrary/src/graph/nodes/
DSplitLayerNode.cpp59 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
113 TensorDescriptor output_descriptor = input_descriptor; in configure_output() local
DDetectionOutputLayerNode.cpp52 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DPriorBoxLayerNode.cpp54 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DDepthToSpaceLayerNode.cpp55 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DReorgLayerNode.cpp57 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DPoolingLayerNode.cpp61 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DStackLayerNode.cpp55 TensorDescriptor output_descriptor = input_descriptors[0]; in compute_output_descriptor() local
DDeconvolutionLayerNode.cpp62 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DConcatenateLayerNode.cpp69 TensorDescriptor output_descriptor = input_descriptors[0]; in compute_output_descriptor() local
DFullyConnectedLayer.cpp102 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DFusedConvolutionBatchNormalizationWithPostOpsNode.cpp95 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DFusedDepthwiseConvolutionBatchNormalizationNode.cpp97 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DDepthwiseConvolutionLayerNode.cpp95 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DConvolutionLayerNode.cpp106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DFusedConvolutionBatchNormalizationNode.cpp106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
DFusedConvolutionWithPostOpNode.cpp106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h374 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
387 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
400 const dnn::BatchDescriptor& output_descriptor, in DoSeparableConvolve()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dlazy_op_runner.h125 const BatchDescriptor& output_descriptor; member
150 const BatchDescriptor& output_descriptor; member
Dstream.h341 const dnn::BatchDescriptor &output_descriptor, in ConvolveWithAlgorithm()
376 const dnn::BatchDescriptor &output_descriptor, in FusedConvolveWithAlgorithm()
398 const dnn::BatchDescriptor &output_descriptor, in ConvolveRunnerFromDesc()
418 const dnn::BatchDescriptor &output_descriptor, in FusedConvolveRunnerFromDesc()
Ddnn.cc137 const dnn::BatchDescriptor& output_descriptor, in ConvolveRunnerFromDesc()
150 const dnn::BatchDescriptor& output_descriptor, in GetFusedConvolveRunners()
166 const dnn::BatchDescriptor& output_descriptor, in FusedConvolveRunnerFromDesc()
Dstream_executor_pimpl.cc298 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetConvolveRunners()
322 const dnn::BatchDescriptor& output_descriptor, in GetFusedConvolveRunners()
341 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetMIOpenConvolveAlgorithms()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor_test.cc67 TensorDescriptor output_descriptor; in TensorBHWCTest() local
114 TensorDescriptor output_descriptor; in TensorBHWDCTest() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h63 se::dnn::BatchDescriptor output_descriptor; member
Dgpu_conv_runner.cc437 BatchDescriptor& output_descriptor = config.output_descriptor; in GetGpuConvConfig() local

12