| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | depthwise_conv_3x3_test_util.cc | 30 TensorFloat32 src_tensor; in DepthwiseConv3x3SimpleWeightsTest() local
68 TensorFloat32 src_tensor; in DepthwiseConv3x3Test() local
|
| H A D | convolution_transposed_thin_test_util.cc | 31 TensorFloat32 src_tensor; in ConvolutionTransposedThinSimpleWeightsTest() local
72 TensorFloat32 src_tensor; in ConvolutionTransposedThinTest() local
|
| H A D | convolution_transposed_test_util.cc | 31 TensorFloat32 src_tensor; in ConvolutionTransposedSimpleWeightsTest() local
72 TensorFloat32 src_tensor; in ConvolutionTransposedTest() local
|
| H A D | prelu_test_util.cc | 30 TensorFloat32 src_tensor; in PReLUAlphaTest() local
61 TensorFloat32 src_tensor; in PReLUHWCAlphaTest() local
|
| H A D | gather_test_util.cc | 30 TensorFloat32 src_tensor; in GatherWidthIntTest() local
70 TensorFloat32 src_tensor; in GatherWidthTest() local
|
| H A D | one_hot_test_util.cc | 26 tflite::gpu::Tensor<BHWC, DataType::INT32> src_tensor; in OneHotTest() local
55 tflite::gpu::Tensor<BHWC, DataType::INT32> src_tensor; in OneHotBatchTest() local
|
| H A D | cumsum_test_util.cc | 28 Tensor<HWC, DataType::FLOAT32> src_tensor; in CumsumHWC() local
114 Tensor<BHWC, DataType::FLOAT32> src_tensor; in CumsumBHWC() local
|
| H A D | strided_slice_test_util.cc | 30 TensorFloat32 src_tensor; in StridedSliceTest() local
|
| H A D | convolution_transposed_3x3_test_util.cc | 30 TensorFloat32 src_tensor; in ConvolutionTransposed3x3Test() local
|
| H A D | depthwise_conv_3x3_stride_h2_test_util.cc | 31 TensorFloat32 src_tensor; in DepthWiseConv3x3StrideH2SimpleWeightsTest() local
|
| H A D | convolution_transposed_4x4_test_util.cc | 31 TensorFloat32 src_tensor; in ConvolutionTransposed4x4SimpleWeightsTest() local
|
| H A D | max_unpooling_test_util.cc | 30 TensorFloat32 src_tensor; in MaxUnpoolingTest() local
|
| H A D | lstm_test_util.cc | 29 TensorFloat32 src_tensor; in LstmTest() local
|
| H A D | resampler_test_util.cc | 31 TensorFloat32 src_tensor; in ResamplerIdentityTest() local
|
| H A D | transpose_test_util.cc | 116 TensorFloat32 src_tensor; in TransposeTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| H A D | call.cc | 133 TfLiteTensor* src_tensor = context->tensors + node->inputs->data[i]; in CopyInputTensorsData() local
151 const TfLiteTensor* src_tensor = in CopyOutputTensorsData() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_quantize_op.cc | 276 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in ComputeScalar() local
340 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in Compute() local
|
| H A D | mkl_lrn_op.cc | 97 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in Compute() local
283 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in SanityCheckInputs() local
|
| H A D | mkl_swish_op.cc | 37 const Tensor& src_tensor = context->input(0); in Compute_Scalar() local
|
| H A D | mkl_fused_mish_op.cc | 36 const Tensor& src_tensor = context->input(0); in Compute_Scalar() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/ |
| H A D | BlitNode.cpp | 29 auto src_tensor = graph->get_tensor(src_); in encode() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | tensor_aten_ops.cpp | 63 at::Tensor src_tensor = src->ToTensor(/*detached=*/true); in copy_() local
|
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ |
| H A D | ClPRelu.cpp | 54 auto src_tensor = const_cast<ITensor *>(tensors.get_const_tensor(TensorType::ACL_SRC_0)); in run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| H A D | resampler_test.cc | 34 TensorRef<BHWC> src_tensor; in ResamplerIdentityTest() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | BatchNormalizationLayerFixture.h | 60 void fill(U &&src_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U &&gamma_tensor) in fill()
|