Home
last modified time | relevance | path

Searched defs:expected_output_shape (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEPadLayerKernel.cpp49 …const TensorShape expected_output_shape = arm_compute::misc::shape_calculator::compute_padded_shap… in validate_arguments() local
190 …const TensorShape expected_output_shape = arm_compute::misc::shape_calculator::compute_padded_shap… in configure() local
H A DNESpaceToBatchLayerKernel.cpp78 …TensorShape expected_output_shape = misc::shape_calculator::compute_space_to_batch_shape(input, bl… in validate_arguments_static() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEReductionOperation.cpp79 …const TensorInfo expected_output_shape = output->clone()->set_tensor_shape(arm_compute::misc::shap… in validate() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLReductionOperation.cpp58 …const TensorInfo expected_output_shape = output->clone()->set_tensor_shape(arm_compute::misc::shap… in validate() local
H A DCLArgMinMaxLayer.cpp66 …const TensorInfo expected_output_shape = output->clone()->set_tensor_shape(arm_compute::misc::shap… in validate() local
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLSpaceToBatchLayerKernel.cpp74 …TensorShape expected_output_shape = misc::shape_calculator::compute_space_to_batch_shape(input, bl… in validate_arguments_static() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClIm2ColKernel.cpp101 …TensorShape expected_output_shape = compute_im2col_conv_shape(src, kernel_dims, conv_info, has_bia… in validate_and_configure_window() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClGemmConv2d.cpp452 …TensorShape expected_output_shape = compute_im2col_conv_shape(src, kernel_dims, conv2d_info.conv_i… in validate() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dtesting_utils.py94 expected_output_shape=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc7243 std::vector<int> expected_output_shape; in TEST_P() member