Home
last modified time | relevance | path

Searched defs:input_descriptor (Results 1 – 25 of 52) sorted by relevance

123

/external/ComputeLibrary/examples/
Dgraph_mobilenet.cpp76 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
121 void create_graph_float(TensorDescriptor &input_descriptor, int model_id) in create_graph_float()
176 void create_graph_qasymm(TensorDescriptor &input_descriptor) in create_graph_qasymm()
Dgraph_mobilenet_v2.cpp68 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
126 void create_graph_float(TensorDescriptor &input_descriptor) in create_graph_float()
262 void create_graph_qasymm8(TensorDescriptor &input_descriptor) in create_graph_qasymm8()
Dgraph_lenet.cpp71 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
Dgraph_vgg19.cpp70 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
Dgraph_vgg16.cpp71 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
Dgraph_alexnet.cpp78 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
Dgraph_mnist.cpp75 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
Dgraph_srcnn955.cpp85 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
Dgraph_squeezenet_v1_1.cpp71 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
Dgraph_vgg_vdsr.cpp86 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
/external/ComputeLibrary/src/graph/nodes/
DFullyConnectedLayer.cpp46 …iptor FullyConnectedLayerNode::compute_weights_descriptor(const TensorDescriptor &input_descriptor, in compute_weights_descriptor()
81 …riptor FullyConnectedLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
DSplitLayerNode.cpp52 …r, Coordinates> SplitLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
112 TensorDescriptor input_descriptor = src->desc(); in configure_output() local
DDetectionOutputLayerNode.cpp47 …DetectionOutputLayerNode::compute_output_descriptor(const TensorDescriptor &input_descript… in compute_output_descriptor()
DSliceLayerNode.cpp52 TensorDescriptor SliceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
DDepthToSpaceLayerNode.cpp47 …riptor DepthToSpaceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, … in compute_output_descriptor()
DPriorBoxLayerNode.cpp47 …orDescriptor PriorBoxLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
DReorgLayerNode.cpp46 TensorDescriptor ReorgLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor in compute_output_descriptor()
DUpsampleLayerNode.cpp51 …orDescriptor UpsampleLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
DPoolingLayerNode.cpp47 …sorDescriptor PoolingLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
DStridedSliceLayerNode.cpp65 …scriptor StridedSliceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
DDeconvolutionLayerNode.cpp47 …criptor DeconvolutionLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h352 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
365 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
/external/ComputeLibrary/tests/validate_examples/
Dgraph_fully_connected.cpp173 …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
/external/tensorflow/tensorflow/stream_executor/
Dstream.h332 const dnn::BatchDescriptor &input_descriptor, in ConvolveWithAlgorithm()
361 const dnn::BatchDescriptor &input_descriptor, in ConvolveWithExecutionPlan()
460 const dnn::BatchDescriptor &input_descriptor, in ConvolveBackwardDataWithExecutionPlan()
488 const dnn::BatchDescriptor &input_descriptor, in ConvolveBackwardDataWithAlgorithm()
515 const dnn::BatchDescriptor &input_descriptor, in ConvolveBackwardFilterWithAlgorithm()
547 const dnn::BatchDescriptor &input_descriptor, in ConvolveBackwardFilterWithExecutionPlan()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h342 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
355 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()

123