| /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ |
| H A D | Im2Col.cpp | 149 const TensorShape src_shape = TensorShape(7U, 27U, 13U); in TEST_SUITE() local
|
| H A D | DirectConvolutionLayer.cpp | 147 const auto src_shape = TensorShape(27U, 13U, 2U); in TEST_SUITE() local
|
| H A D | DepthwiseConvolutionLayerNative.cpp | 118 TensorShape src_shape{ depth, input_size.x(), input_size.y() }; in TEST_SUITE() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuCopyKernel.cpp | 69 const TensorShape src_shape = src->tensor_shape(); in validate_and_configure_window_with_padding() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | PoolingLayerFixture.h | 214 void setup(TensorShape src_shape, PoolingLayerInfo pool_info, DataType data_type) in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| H A D | test_util.cc | 145 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/heuristics/dwc_native/ |
| H A D | ClDWCNativeDefaultConfigValhall.cpp | 143 const TensorShape src_shape = src->tensor_shape(); in configure_G78_f16() local
|
| H A D | ClDWCNativeDefaultConfigBifrost.cpp | 118 const TensorShape src_shape = src->tensor_shape(); in configure_f16() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| H A D | testing_util.cc | 33 auto src_shape = src_cpu[i]->GetBHWDCShape(); in ExecuteGPUOperation() local
|
| /aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/ |
| H A D | GpuKernelComponentGroup.cpp | 192 const auto &src_shape = src_tensor->tensor_shape(); in finalize() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | neon_cnn.cpp | 68 const TensorShape src_shape(width_src_image, height_src_image, ifm_src_img); in do_setup() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_eltwise_activation_base_op.h | 249 TensorShape src_shape = src_tensor.shape(); in Compute() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/internal/ |
| H A D | CpuPool2dAssemblyWrapperKernel.cpp | 161 const auto src_shape = src->info()->tensor_shape(); in run_op() local
|
| H A D | CpuDepthwiseConv2dAssemblyWrapperKernel.cpp | 323 const auto src_shape = src->info()->tensor_shape(); in run_op() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | elementwise.cc | 380 std::string GetReadBroadcastedValueCode(const BHWC& src_shape, in GetReadBroadcastedValueCode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape_tree.h | 270 const Shape& src_shape = ShapeUtil::GetSubshape(other.shape(), src_index); in CopySubtreeFrom() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | model_builder.cc | 594 BHWC src_shape, weights_shape; in Parse() local
638 BHWC src_shape; in ReadAttributes() local
673 BHWC src_shape, dst_shape; in ResolveGroupedConvolution() local
1867 auto src_shape = graph->FindInputs(node->id)[0]->tensor.shape; in Parse() local
|
| H A D | gpu_model.cc | 121 const auto src_shape = inputs[i]->tensor.shape; in IsAssociativeLinkableOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | shape_refiner.cc | 457 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ragged_tensor_to_tensor_op.cc | 499 const auto& src_shape = default_value_tensor.shape(); in SetOutput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| H A D | operation_selector.cc | 160 const BHWC& src_shape, const OHWI& weights_shape, in AddDynamicConv()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_verifier.cc | 1512 const Shape& src_shape = ShapeUtil::GetTupleElementShape(operand_shape, 1); in HandleCopyDone() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 4310 def helper(src_shape, dst_shape, src_dtype, dst_dtype): argument
7456 def helper(shape, dim, idx_shape, src_shape, idx_dtype=torch.int64, do_add=True): argument
7563 def helper(shape, dim, idx_shape, src_shape, idx_dtype=torch.int64, reduce_str="sum"): argument
|