Home
last modified time | relevance | path

Searched defs:input_descriptor (Results 1 – 25 of 53) 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
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()
Dgraph_mobilenet_v2.cpp68 …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()
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_alexnet.cpp78 …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_srcnn955.cpp85 …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
Dgraph_squeezenet_v1_1.cpp71 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
Dgraph_squeezenet.cpp71 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
/external/ComputeLibrary/src/graph/nodes/
DFullyConnectedLayer.cpp55 …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()
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()
DPriorBoxLayerNode.cpp47 …orDescriptor PriorBoxLayerNode::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()
DReorgLayerNode.cpp46 TensorDescriptor ReorgLayerNode::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()
DConcatenateLayerNode.cpp76 for(auto &input_descriptor : input_descriptors) in compute_output_descriptor() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h364 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
377 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/compiler/xla/stream_executor/
Dlazy_op_runner.h123 const BatchDescriptor& input_descriptor; member
147 const BatchDescriptor& input_descriptor; member
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h368 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
381 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()

123