Searched defs:mul_tensor (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| H A D | fuse_mul_to_conv_test.cc | 55 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
96 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
149 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
170 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
192 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
213 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
233 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
254 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
276 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
297 Tensor<Linear, DataType::FLOAT32> mul_tensor; in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | frozen_conv_folding.cpp | 337 Tensor mul_tensor = resizeConstantScalarOrTensorToShape( in FoldFrozenConvMulOrDiv() local
364 auto mul_tensor = resizeConstantScalarOrTensorToShape( in FoldFrozenConvMulOrDiv() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_batch_matmul_op.cc | 168 const Tensor& mul_tensor = ctx->input(2); in Compute() local
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_shape_analysis.cpp | 432 auto mul_tensor = in TEST() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | BinaryOp.cpp | 503 static Tensor mul_tensor(const Tensor& self_arg, const Tensor& other_arg) { in mul_tensor() function
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 938 const auto mul_tensor = R"JIT( in TEST() local
|