Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
H A 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
H A Dadd_quant_adjustments_test.cc57 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
101 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/
H A 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
/aosp_15_r20/external/ComputeLibrary/src/core/
H A 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
/aosp_15_r20/external/ComputeLibrary/src/common/
H A DTensorPack.cpp44 AclStatus TensorPack::add_tensor(ITensorV2 *tensor, int32_t slot_id) in add_tensor() function in arm_compute::TensorPack
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dvariant_tensor_data.h129 Tensor* VariantTensorData::add_tensor(TensorConstructorArgs&&... args) { in add_tensor() function
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_autodiff.py13 def fn(tensor_list: List[torch.Tensor], add_tensor): argument
H A Dtest_freezing.py2199 use_bias, module, tracing, op, scalar, add_tensor, expect_success argument
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/
H A DArgumentPack.h104 void add_tensor(Id id, T *tensor) in add_tensor() function
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dtorchbind_impls.py88 def add_tensor(self, z): member in register_fake_classes.FakeFoo
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_binary_folding.py37 def test_conv_fusion(use_bias, module, op, scalar, add_tensor, expect_success): argument
H A Dtest_torchinductor.py8279 add_tensor, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_batch_matmul_op.cc173 const Tensor& add_tensor = ctx->input(3); in Compute() local
H A Dmkl_matmul_op_fused.cc157 const Tensor& add_tensor = MklGetInput(ctx, kInputIndex_Add); in Compute() local
H A Dmkl_conv_ops.cc1113 const Tensor& add_tensor = MklGetInput(context, kInputIndex_Add); in AllocateOutputTensor() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_static_runtime.py133 def add_tensor(input1, input2): function
/aosp_15_r20/external/pytorch/torch/utils/tensorboard/
H A Dwriter.py427 def add_tensor( member in SummaryWriter
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DBinaryOp.cpp439 static Tensor add_tensor( in add_tensor() function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/util/
Dsaved_tensor_slice.pb.h993 inline ::tensorflow::SavedSliceMeta* SavedTensorSliceMeta::add_tensor() { in add_tensor() function
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_custom_class_registrations.cpp49 at::Tensor add_tensor(at::Tensor z) { in add_tensor() function
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_unflatten.py619 def add_tensor(self, z): member in TestUnflatten.test_unflatten_constant_obj.FakeFoo
H A Dtest_torchbind.py79 def add_tensor(self, z): member in TestExportTorchbind.setUp.FakeFoo
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dattr_value.pb.h1235 inline ::tensorflow::TensorProto* AttrValue_ListValue::add_tensor() { in add_tensor() function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Deager_service.pb.h5506 inline ::tensorflow::TensorProto* QueueResponse::add_tensor() { in add_tensor() function
6467 inline ::tensorflow::TensorProto* RunComponentFunctionResponse::add_tensor() { in add_tensor() function
Dmaster.pb.h4262 inline ::tensorflow::NamedTensorProto* RunStepResponse::add_tensor() { in add_tensor() function