| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_mobilenet.cpp | 76 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
122 void create_graph_float(TensorDescriptor &input_descriptor, int model_id) in create_graph_float()
177 void create_graph_qasymm(TensorDescriptor &input_descriptor) in create_graph_qasymm()
|
| H A D | graph_mobilenet_v2.cpp | 68 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
127 void create_graph_float(TensorDescriptor &input_descriptor) in create_graph_float()
263 void create_graph_qasymm8(TensorDescriptor &input_descriptor) in create_graph_qasymm8()
|
| H A D | graph_ssd_mobilenet.cpp | 79 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
216 void create_graph_float(TensorDescriptor &input_descriptor) in create_graph_float()
459 void create_graph_qasymm(TensorDescriptor &input_descriptor) in create_graph_qasymm()
|
| H A D | graph_lenet.cpp | 71 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
|
| H A D | graph_vgg19.cpp | 70 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
|
| H A D | graph_alexnet.cpp | 78 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
|
| H A D | graph_vgg16.cpp | 71 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
|
| H A D | graph_srcnn955.cpp | 85 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 439 miopenTensorDescriptor_t input_descriptor) { in FindOrCreate()
1085 const miopenTensorDescriptor_t input_descriptor) in ScopedFusionPlanBase()
1105 miopenStatus_t Execute(miopenTensorDescriptor_t input_descriptor, in Execute()
1285 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1365 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1401 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationInference()
1471 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1501 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationForward()
1571 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1602 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationBackward()
[all …]
|
| H A D | rocm_dnn.h | 368 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
381 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | FullyConnectedLayer.cpp | 55 …iptor FullyConnectedLayerNode::compute_weights_descriptor(const TensorDescriptor &input_descriptor, in compute_weights_descriptor()
90 …riptor FullyConnectedLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
|
| H A D | SplitLayerNode.cpp | 52 …r, Coordinates> SplitLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
112 TensorDescriptor input_descriptor = src->desc(); in configure_output() local
|
| H A D | DetectionOutputLayerNode.cpp | 47 …DetectionOutputLayerNode::compute_output_descriptor(const TensorDescriptor &input_descript… in compute_output_descriptor()
|
| H A D | DepthToSpaceLayerNode.cpp | 47 …riptor DepthToSpaceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, … in compute_output_descriptor()
|
| H A D | PriorBoxLayerNode.cpp | 47 …orDescriptor PriorBoxLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
|
| H A D | SliceLayerNode.cpp | 52 TensorDescriptor SliceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
|
| H A D | ReorgLayerNode.cpp | 46 TensorDescriptor ReorgLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor… in compute_output_descriptor()
|
| H A D | PoolingLayerNode.cpp | 47 …sorDescriptor PoolingLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
|
| H A D | StridedSliceLayerNode.cpp | 65 …scriptor StridedSliceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.h | 364 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
377 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validate_examples/ |
| H A D | graph_fully_connected.cpp | 173 …const TensorDescriptor input_descriptor = TensorDescriptor(input_shape, _params.data_type, _para… in create_tensors() local
191 …const TensorDescriptor input_descriptor = TensorDescriptor(input_shape, _params.data_type, _params… in output_shape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | lazy_op_runner.h | 123 const BatchDescriptor& input_descriptor; member
147 const BatchDescriptor& input_descriptor; member
|
| H A D | stream.h | 337 dnn::ConvolutionKind kind, const dnn::BatchDescriptor &input_descriptor, in ConvolveWithAlgorithm()
396 const dnn::BatchDescriptor &input_descriptor, in ConvolveRunnerFromDesc()
415 const dnn::BatchDescriptor &input_descriptor, in FusedConvolveRunnerFromDesc()
|
| H A D | dnn.cc | 135 dnn::DataType output_type, const dnn::BatchDescriptor& input_descriptor, in ConvolveRunnerFromDesc()
147 const dnn::BatchDescriptor& input_descriptor, in GetFusedConvolveRunners()
163 const dnn::BatchDescriptor& input_descriptor, in FusedConvolveRunnerFromDesc()
|
| H A D | stream_executor_pimpl.cc | 296 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in GetConvolveRunners()
319 const dnn::BatchDescriptor& input_descriptor, in GetFusedConvolveRunners()
339 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in GetMIOpenConvolveAlgorithms()
|