Home
last modified time | relevance | path

Searched defs:add_tensor (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dfuse_add_to_conv_test.cc55 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
98 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
119 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
141 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
162 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
180 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
Dadd_quant_adjustments_test.cc57 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
101 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
/external/ComputeLibrary/src/core/
DITensorPack.cpp39 void ITensorPack::add_tensor(int id, ITensor *tensor) in add_tensor() function in arm_compute::ITensorPack
44 void ITensorPack::add_tensor(int id, const ITensor *tensor) in add_tensor() function in arm_compute::ITensorPack
/external/executorch/backends/vulkan/runtime/graph/
DComputeGraph.cpp237 ValueRef ComputeGraph::add_tensor( in add_tensor() function in vkcompute::ComputeGraph
256 ValueRef ComputeGraph::add_tensor( in add_tensor() function in vkcompute::ComputeGraph
269 ValueRef ComputeGraph::add_tensor( in add_tensor() function in vkcompute::ComputeGraph
292 ValueRef ComputeGraph::add_tensor( in add_tensor() function in vkcompute::ComputeGraph
300 ValueRef ComputeGraph::add_tensor(const vkapi::VulkanImage& image) { in add_tensor() function in vkcompute::ComputeGraph
/external/ComputeLibrary/src/common/
DTensorPack.cpp44 AclStatus TensorPack::add_tensor(ITensorV2 *tensor, int32_t slot_id) in add_tensor() function in arm_compute::TensorPack
/external/tensorflow/tensorflow/core/framework/
Dvariant_tensor_data.h129 Tensor* VariantTensorData::add_tensor(TensorConstructorArgs&&... args) { in add_tensor() function
/external/pytorch/test/jit/
Dtest_autodiff.py13 def fn(tensor_list: List[torch.Tensor], add_tensor): argument
Dtest_freezing.py2199 use_bias, module, tracing, op, scalar, add_tensor, expect_success argument
/external/ComputeLibrary/src/dynamic_fusion/sketch/
DArgumentPack.h104 void add_tensor(Id id, T *tensor) in add_tensor() function
/external/pytorch/torch/testing/_internal/
Dtorchbind_impls.py88 def add_tensor(self, z): member in register_fake_classes.FakeFoo
/external/pytorch/test/inductor/
Dtest_binary_folding.py37 def test_conv_fusion(use_bias, module, op, scalar, add_tensor, expect_success): argument
Dtest_torchinductor.py8279 add_tensor, argument
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_batch_matmul_op.cc173 const Tensor& add_tensor = ctx->input(3); in Compute() local
Dmkl_matmul_op_fused.cc157 const Tensor& add_tensor = MklGetInput(ctx, kInputIndex_Add); in Compute() local
Dmkl_conv_ops.cc1113 const Tensor& add_tensor = MklGetInput(context, kInputIndex_Add); in AllocateOutputTensor() local
/external/pytorch/test/
Dtest_static_runtime.py133 def add_tensor(input1, input2): function
/external/pytorch/torch/utils/tensorboard/
Dwriter.py427 def add_tensor( member in SummaryWriter
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DBinaryOp.cpp439 static Tensor add_tensor( in add_tensor() function
/external/pytorch/test/cpp/jit/
Dtest_custom_class_registrations.cpp49 at::Tensor add_tensor(at::Tensor z) { in add_tensor() function
/external/pytorch/test/export/
Dtest_unflatten.py619 def add_tensor(self, z): member in TestUnflatten.test_unflatten_constant_obj.FakeFoo
Dtest_torchbind.py79 def add_tensor(self, z): member in TestExportTorchbind.setUp.FakeFoo