Home
last modified time | relevance | path

Searched refs:current_output_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DInPlaceOperationMutator.cpp124 auto current_output_tensor = node->output(0); in try_in_place_depthwiseconv() local
125 ARM_COMPUTE_ERROR_ON(current_output_tensor == nullptr); in try_in_place_depthwiseconv()
126 const auto out_shape = current_output_tensor->desc().shape; in try_in_place_depthwiseconv()
127 const auto qinfo_out = current_output_tensor->desc().quant_info; in try_in_place_depthwiseconv()
146 set_new_output_and_inherit_accessor(node, current_output_tensor, input_tensor); in try_in_place_depthwiseconv()
179 auto current_output_tensor = node->output(0); in try_in_place_elementwise() local
180 ARM_COMPUTE_ERROR_ON(current_output_tensor == nullptr); in try_in_place_elementwise()
181 const auto qinfo_out = current_output_tensor->desc().quant_info; in try_in_place_elementwise()
185 …&& (input0_tensor->desc().data_type == current_output_tensor->desc().data_type) && (input0_tensor-… in try_in_place_elementwise()
187 …&& (input1_tensor->desc().data_type == current_output_tensor->desc().data_type) && (input1_tensor-… in try_in_place_elementwise()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DINode.cpp92 auto current_output_tensor = output_edge->tensor(); in set_output_tensor() local
93 current_output_tensor->unbind_edge(output_edge->id()); in set_output_tensor()