| /external/ComputeLibrary/src/graph/nodes/ |
| D | FullyConnectedLayer.cpp | 55 …iptor FullyConnectedLayerNode::compute_weights_descriptor(const TensorDescriptor &input_descriptor, in compute_weights_descriptor() argument
61 unsigned int num_dimensions = input_descriptor.shape.num_dimensions(); in compute_weights_descriptor()
69 num_weights *= input_descriptor.shape[i]; in compute_weights_descriptor()
72 TensorDescriptor weights_descriptor = input_descriptor; in compute_weights_descriptor()
90 …riptor FullyConnectedLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
95 unsigned int batches = input_descriptor.shape[1]; in compute_output_descriptor()
96 if(input_descriptor.shape.num_dimensions() > 2) in compute_output_descriptor()
98 batches = input_descriptor.shape[3]; in compute_output_descriptor()
102 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | SplitLayerNode.cpp | 52 …r, Coordinates> SplitLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
56 int num_dimension = static_cast<int32_t>(input_descriptor.shape.num_dimensions()); in compute_output_descriptor()
59 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
60 int split_size = input_descriptor.shape[tmp_axis] / num_splits; in compute_output_descriptor()
71 split_size = input_descriptor.shape[tmp_axis]; in compute_output_descriptor()
112 TensorDescriptor input_descriptor = src->desc(); in configure_output() local
113 TensorDescriptor output_descriptor = input_descriptor; in configure_output()
119 …int split_size = (_size_splits.empty()) ? (input_descriptor.shape[tmp_axis] / _num_splits) : _size… in configure_output()
122 split_size = input_descriptor.shape[tmp_axis]; in configure_output()
|
| D | ReorgLayerNode.cpp | 46 TensorDescriptor ReorgLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor… in compute_output_descriptor() argument
48 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
49 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
50 …const unsigned int input_channel = get_dimension_size(input_descriptor, DataLayoutDimension::CHANN… in compute_output_descriptor()
56 const DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
57 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | PoolingLayerNode.cpp | 47 …sorDescriptor PoolingLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
53 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH); in compute_output_descriptor()
54 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGHT… in compute_output_descriptor()
60 const DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
61 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | DepthwiseConvolutionLayerNode.cpp | 78 … DepthwiseConvolutionLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
86 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDT… in compute_output_descriptor()
87 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIG… in compute_output_descriptor()
88 …const unsigned int input_channels = get_dimension_size(input_descriptor, DataLayoutDimension::CHAN… in compute_output_descriptor()
94 const DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
95 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | DeconvolutionLayerNode.cpp | 47 …criptor DeconvolutionLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
54 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
55 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
61 const DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
62 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | DepthToSpaceLayerNode.cpp | 47 …ToSpaceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, int block_sh… in compute_output_descriptor() argument
51 TensorShape input_shape = input_descriptor.shape; in compute_output_descriptor()
52 DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
55 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | DetectionOutputLayerNode.cpp | 47 …tectionOutputLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
50 …nsigned int max_size = info.keep_top_k() * ((input_descriptor.shape.num_dimensions() > 1) ? input_… in compute_output_descriptor()
52 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | FusedConvolutionBatchNormalizationWithPostOpsNode.cpp | 80 …chNormalizationWithPostOpsNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
87 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
88 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
94 const DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
95 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | FusedDepthwiseConvolutionBatchNormalizationNode.cpp | 81 …volutionBatchNormalizationNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
89 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDT… in compute_output_descriptor()
90 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIG… in compute_output_descriptor()
91 …const unsigned int input_channels = get_dimension_size(input_descriptor, DataLayoutDimension::CHAN… in compute_output_descriptor()
97 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | PriorBoxLayerNode.cpp | 47 …orDescriptor PriorBoxLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
50 …const unsigned int layer_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH); in compute_output_descriptor()
51 …const unsigned int layer_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGHT… in compute_output_descriptor()
54 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | ConvolutionLayerNode.cpp | 91 …escriptor ConvolutionLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
98 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
99 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
105 const DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | FusedConvolutionBatchNormalizationNode.cpp | 91 …volutionBatchNormalizationNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
98 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
99 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
105 const DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | FusedConvolutionWithPostOpNode.cpp | 91 …FusedConvolutionWithPostOpNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
98 …const unsigned int input_width = get_dimension_size(input_descriptor, DataLayoutDimension::WIDTH… in compute_output_descriptor()
99 …const unsigned int input_height = get_dimension_size(input_descriptor, DataLayoutDimension::HEIGH… in compute_output_descriptor()
105 const DataLayout data_layout = input_descriptor.layout; in compute_output_descriptor()
106 TensorDescriptor output_descriptor = input_descriptor; in compute_output_descriptor()
|
| D | SliceLayerNode.cpp | 52 TensorDescriptor SliceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
57 TensorDescriptor output_desc = input_descriptor; in compute_output_descriptor()
58 … = arm_compute::misc::shape_calculator::compute_slice_shape(input_descriptor.shape, starts… in compute_output_descriptor()
|
| D | StridedSliceLayerNode.cpp | 65 …scriptor StridedSliceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, in compute_output_descriptor() argument
73 TensorDescriptor output_desc = input_descriptor; in compute_output_descriptor()
74 …output_desc.shape = compute_strided_slice_output_shape(input_descriptor.shape, starts, … in compute_output_descriptor()
|
| D | ConcatenateLayerNode.cpp | 76 for(auto &input_descriptor : input_descriptors) in compute_output_descriptor() local
78 shapes.emplace_back(&input_descriptor.shape); in compute_output_descriptor()
|
| /external/ComputeLibrary/tests/validate_examples/ |
| D | graph_fully_connected.cpp | 173 …const TensorDescriptor input_descriptor = TensorDescriptor(input_shape, _params.data_type, _para… in create_tensors() local
174 …scriptor weights_descriptor = FullyConnectedLayerNode::compute_weights_descriptor(input_descriptor, in create_tensors()
178 …tput_desciptor = FullyConnectedLayerNode::compute_output_descriptor(input_descriptor, _params.full… in create_tensors()
181 …src = SimpleTensor<D> { input_descriptor.shape, _params.data_type, 1, input_descriptor.quant_i… in create_tensors()
191 …const TensorDescriptor input_descriptor = TensorDescriptor(input_shape, _params.data_type, _params… in output_shape() local
192 …tput_desciptor = FullyConnectedLayerNode::compute_output_descriptor(input_descriptor, _params.full… in output_shape()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | lazy_op_runner.h | 123 const BatchDescriptor& input_descriptor; member
134 config.input_descriptor, config.filter_descriptor, in RunnerFromAlgorithmDesc()
147 const BatchDescriptor& input_descriptor; member
161 config.leakyrelu_alpha, config.input_descriptor, in RunnerFromAlgorithmDesc()
|
| /external/ComputeLibrary/examples/ |
| D | graph_mobilenet.cpp | 76 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
85 create_graph_float(input_descriptor, model_id); in do_setup()
89 create_graph_qasymm(input_descriptor); in do_setup()
122 void create_graph_float(TensorDescriptor &input_descriptor, int model_id) in create_graph_float() argument
139 graph << InputLayer(input_descriptor, in create_graph_float()
177 void create_graph_qasymm(TensorDescriptor &input_descriptor) in create_graph_qasymm() argument
236 graph << InputLayer(input_descriptor.set_quantization_info(in_quant_info), in create_graph_qasymm()
|
| D | graph_mobilenet_v2.cpp | 68 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
77 create_graph_float(input_descriptor); in do_setup()
81 create_graph_qasymm8(input_descriptor); in do_setup()
127 void create_graph_float(TensorDescriptor &input_descriptor) in create_graph_float() argument
144 …graph << InputLayer(input_descriptor, get_input_accessor(common_params, std::move(preprocessor), f… in create_graph_float()
263 void create_graph_qasymm8(TensorDescriptor &input_descriptor) in create_graph_qasymm8() argument
351 graph << InputLayer(input_descriptor.set_quantization_info(in_quant_info), in create_graph_qasymm8()
|
| D | graph_lenet.cpp | 71 …TensorDescriptor input_descriptor = TensorDescriptor(tensor_shape, common_params.data_type).set_l… in do_setup() local
79 << InputLayer(input_descriptor, get_input_accessor(common_params)) in do_setup()
|
| /external/ComputeLibrary/arm_compute/graph/nodes/ |
| D | FullyConnectedLayerNode.h | 74 static TensorDescriptor compute_weights_descriptor(const TensorDescriptor &input_descriptor,
88 static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 104 params.config->input_descriptor, in RunGpuConvUnfused()
164 params.config->input_descriptor, in RunGpuConvForwardActivation()
393 BatchDescriptor& input_descriptor = config.input_descriptor; in GetGpuConvConfig() local
394 input_descriptor = BatchDescriptor(effective_num_dimensions); in GetGpuConvConfig()
395 input_descriptor.set_layout(input_dl) in GetGpuConvConfig()
402 input_descriptor.set_spatial_dim( in GetGpuConvConfig()
452 input_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 439 miopenTensorDescriptor_t input_descriptor) { in FindOrCreate() argument
450 input_descriptor); in FindOrCreate()
1085 const miopenTensorDescriptor_t input_descriptor) in ScopedFusionPlanBase() argument
1105 miopenStatus_t Execute(miopenTensorDescriptor_t input_descriptor, in Execute() argument
1110 miopen_handle_, fusion_plan_, input_descriptor, input_data, in Execute()
1285 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument
1291 input_descriptor) { in ScopedFusionPlanConvolutionBiasActivation()
1293 miopen_handle, input_descriptor, filter_descriptor, conv_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1297 hash, &fusion_plan_, miopenVerticalFusion, input_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
1365 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue() argument
[all …]
|