Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A 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()
H A Dcuda_dnn.cc3508 const dnn::BatchDescriptor& output_descriptor, in GetCudnnOperationGraph()
3635 const dnn::BatchDescriptor& output_descriptor, in GetCudnnFusedOperationGraph()
3960 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
4261 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
4717 const dnn::BatchDescriptor& output_descriptor, in GetConvolveRunners()
4821 const dnn::BatchDescriptor& output_descriptor, in ConvolveRunnerFromDesc()
5070 const dnn::BatchDescriptor& output_descriptor, in FusedConvolveRunnerFromDesc()
5145 const dnn::BatchDescriptor& output_descriptor, in GetFusedConvolveRunners()
5764 const dnn::BatchDescriptor& output_descriptor, DeviceMemoryBase output_data, in DoFusedConvolve()
6174 const dnn::BatchDescriptor& output_descriptor, dnn::DataType element_type) { in GetTensorSplits()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A 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()
H A Drocm_dnn.cc1107 miopenTensorDescriptor_t output_descriptor, in Execute()
2907 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
2957 BatchDescriptor output_descriptor, in RocmConvRunner()
3116 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
3151 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetConvolveRunners()
3191 const dnn::BatchDescriptor& output_descriptor, in ConvolveRunnerFromDesc()
3216 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetMIOpenConvolveAlgorithms()
3238 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3447 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
3806 const dnn::BatchDescriptor& output_descriptor, DeviceMemoryBase output_data, in DoFusedConvolve()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/
H A DSplitLayerNode.cpp59 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
113 TensorDescriptor output_descriptor = input_descriptor; in configure_output() local
H A DDetectionOutputLayerNode.cpp52 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DDepthToSpaceLayerNode.cpp55 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DPriorBoxLayerNode.cpp54 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DReorgLayerNode.cpp57 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DPoolingLayerNode.cpp61 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DStackLayerNode.cpp55 TensorDescriptor output_descriptor = input_descriptors[0]; in compute_output_descriptor() local
H A DDeconvolutionLayerNode.cpp62 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DConcatenateLayerNode.cpp69 TensorDescriptor output_descriptor = input_descriptors[0]; in compute_output_descriptor() local
H A DFullyConnectedLayer.cpp102 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DFusedConvolutionBatchNormalizationWithPostOpsNode.cpp95 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DDepthwiseConvolutionLayerNode.cpp95 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DFusedDepthwiseConvolutionBatchNormalizationNode.cpp97 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DConvolutionLayerNode.cpp106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DFusedConvolutionBatchNormalizationNode.cpp106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
H A DFusedConvolutionWithPostOpNode.cpp106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A 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()
H A Dlazy_op_runner.h125 const BatchDescriptor& output_descriptor; member
150 const BatchDescriptor& output_descriptor; member
H A 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()
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dtensor_test.cc67 TensorDescriptor output_descriptor; in TensorBHWCTest() local
114 TensorDescriptor output_descriptor; in TensorBHWDCTest() local

12