| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLStackLayerKernel.cpp | 45 … ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsigned int num_tensors, const IT… in validate_arguments()
64 …ate_and_configure_window(ITensorInfo *input, unsigned int axis, unsigned int num_tensors, ITensorI… in validate_and_configure_window()
82 …st ICLTensor *input, unsigned int axis, unsigned int idx_input, unsigned int num_tensors, ICLTenso… in configure()
87 …st ICLTensor *input, unsigned int axis, unsigned int idx_input, unsigned int num_tensors, ICLTenso… in configure()
115 … ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsigned int num_tensors, const IT… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NEStackLayerKernel.cpp | 44 … ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsigned int num_tensors, const IT… in validate_arguments()
63 …ate_and_configure_window(ITensorInfo *input, unsigned int axis, unsigned int num_tensors, ITensorI… in validate_and_configure_window()
92 …onst ITensor *input, unsigned int axis, unsigned int idx_input, unsigned int num_tensors, ITensor … in configure()
109 … ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsigned int num_tensors, const IT… in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scoped_allocator_ops_test.cc | 129 const string& name, int32_t id, int32_t num_tensors) { in BuildNodeDef()
146 int32_t num_tensors) { in BuildNodeDefWithReshape()
163 const string& name, int32_t id, int32_t num_tensors) { in MakeOp()
263 const string& name, int32_t id, int32_t num_tensors, in BuildNodeDef()
279 int32_t id, int32_t num_tensors, in MakeOp()
|
| H A D | save_restore_v2_ops.cc | 47 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in ValidateInputs() local
106 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in Compute() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | StackLayerFixture.h | 56 void setup(TensorShape shape_src, int axis, DataType data_type, int num_tensors) in setup()
69 TensorType compute_target(TensorShape shape_src, int axis, DataType data_type, int num_tensors) in compute_target()
113 …r<T> compute_reference(const TensorShape &shape_src, int axis, DataType data_type, int num_tensors) in compute_reference()
|
| H A D | ConcatenateLayerFixture.h | 61 const int num_tensors = num_dis(gen); in setup() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | python_api_parameter_converter.cc | 268 Py_ssize_t num_tensors = PyList_GET_SIZE(tensor_list); in ConvertInputsWithTypeAttr() local
290 Py_ssize_t num_tensors = PyList_GET_SIZE(tensor_list); in ConvertInputsWithTypeAttr() local
331 Py_ssize_t num_tensors = PyList_GET_SIZE(params[input.tensor_list_params[0]]); in ConvertInputsWithTypeListAttr() local
415 int num_tensors = PyList_GET_SIZE(params[inputs.tensor_list_params[j]]); in InferLengthAttributes() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_flatten.cpp | 68 for (size_t i = 0, num_tensors = tensors.size(); i < num_tensors; ++i) in reorder_tensors_like() local
116 for (size_t i = 0, num_tensors = tensors.size(); i < num_tensors; ++i) { in unflatten_sparse_tensors() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | autograd.cpp | 29 size_t num_tensors = outputs.size(); in _make_grads() local
98 size_t num_tensors = outputs.size(); in run_backward() local
|
| H A D | variable.h | 346 virtual size_t num_tensors() const { in num_tensors() function
380 size_t num_tensors() const override { in num_tensors() function
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | common_utils.py | 154 def get_bdim_choices(num_tensors): argument
172 num_tensors, _, running_mean=None, running_var=None, *args argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device_util.h | 290 virtual size_t num_tensors() const { return tensor()->num_tensors(); } in num_tensors() function
479 size_t num_tensors() const override { return 3 * indices()->num_tensors(); } in num_tensors() function
|
| H A D | dtensor_sharded_prefix_kernel.cc | 77 const int num_tensors = static_cast<int>(tensor_names->NumElements()); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | message_wrappers.cc | 823 size_t InMemoryRunStepResponse::num_tensors() const { return tensors_.size(); } in num_tensors() function in tensorflow::InMemoryRunStepResponse
868 size_t OwnedProtoRunStepResponse::num_tensors() const { in num_tensors() function in tensorflow::OwnedProtoRunStepResponse
924 size_t NonOwnedProtoRunStepResponse::num_tensors() const { in num_tensors() function in tensorflow::NonOwnedProtoRunStepResponse
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/UNIT/ |
| H A D | TensorPackFixture.h | 169 const size_t num_tensors = 256; in setup() local
|
| H A D | TensorFixture.h | 170 const unsigned int num_tensors = 1024; in setup() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | StackLayerNode.cpp | 58 const unsigned int num_tensors = input_descriptors.size(); in compute_output_descriptor() local
|
| /aosp_15_r20/external/ComputeLibrary/src/c/ |
| H A D | AclTensorPack.cpp | 80 …rs(AclTensorPack external_pack, AclTensor *external_tensors, int32_t *slot_ids, size_t num_tensors) in AclPackTensors()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | StackLayer.cpp | 53 const int num_tensors = in.size(); in stack_layer() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | portable_tensor.h | 38 int num_tensors = tensor_list.size; in VectorOfTensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/ |
| H A D | parallel_device_lib.h | 190 size_t num_tensors() const { return tensors_.size(); } in num_tensors() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | memory_management.cc | 47 size_t num_tensors = obj_assignment.object_ids.size(); in ObjectsToOffsets() local
|
| /aosp_15_r20/external/executorch/extension/evalue_util/test/ |
| H A D | print_evalue_test.cpp | 459 void expect_tensor_list_output(size_t num_tensors, const char* expected) { in expect_tensor_list_output()
541 size_t num_tensors, in expect_list_optional_tensor_output()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorShape.cu | 463 auto num_tensors = tensors.size(); in get_chunk_size() local
497 auto num_tensors = tensors.size(); in get_chunk_cat_metadata() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ForeachUtils.h | 293 const auto num_tensors = nested_tensorlist[0].size(); in _group_tensors_by_first_tensors_device_and_dtype() local
|