Home
last modified time | relevance | path

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

123

/external/ComputeLibrary/src/core/CL/kernels/
DCLStackLayerKernel.cpp45 … 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()
DCLArgMinMaxLayerKernel.cpp173 const unsigned int num_tensors = _prev_output != nullptr ? 3 : 2; in run() local
/external/ComputeLibrary/tests/validation/fixtures/
DStackLayerFixture.h56 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()
DConcatenateLayerFixture.h61 const int num_tensors = num_dis(gen); in setup() local
/external/ComputeLibrary/src/core/NEON/kernels/
DNEStackLayerKernel.cpp44 … 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()
/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops_test.cc129 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()
Dsave_restore_v2_ops.cc47 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
/external/pytorch/torch/csrc/utils/
Dtensor_flatten.cpp68 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
/external/tensorflow/tensorflow/python/framework/
Dpython_api_parameter_converter.cc268 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
/external/pytorch/torch/csrc/autograd/
Dautograd.cpp29 size_t num_tensors = outputs.size(); in _make_grads() local
98 size_t num_tensors = outputs.size(); in run_backward() local
Dvariable.h346 virtual size_t num_tensors() const { in num_tensors() function
380 size_t num_tensors() const override { in num_tensors() function
/external/ComputeLibrary/tests/validation/fixtures/UNIT/
DTensorPackFixture.h169 const size_t num_tensors = 256; in setup() local
DTensorFixture.h170 const unsigned int num_tensors = 1024; in setup() local
/external/ComputeLibrary/src/graph/nodes/
DStackLayerNode.cpp58 const unsigned int num_tensors = input_descriptors.size(); in compute_output_descriptor() local
/external/ComputeLibrary/tests/validation/reference/
DStackLayer.cpp53 const int num_tensors = in.size(); in stack_layer() local
/external/ComputeLibrary/src/c/
DAclTensorPack.cpp80 …rs(AclTensorPack external_pack, AclTensor *external_tensors, int32_t *slot_ids, size_t num_tensors) in AclPackTensors()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dportable_tensor.h38 int num_tensors = tensor_list.size; in VectorOfTensors() local
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.h190 size_t num_tensors() const { return tensors_.size(); } in num_tensors() function
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_sharded_prefix_kernel.cc77 const int num_tensors = static_cast<int>(tensor_names->NumElements()); in Compute() local
Ddtensor_device_util.h290 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
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmemory_management.cc47 size_t num_tensors = obj_assignment.object_ids.size(); in ObjectsToOffsets() local
/external/pytorch/aten/src/ATen/native/
DForeachUtils.h293 const auto num_tensors = nested_tensorlist[0].size(); in _group_tensors_by_first_tensors_device_and_dtype() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dgreedy_by_size_assignment.cc72 const size_t num_tensors = usage_records.size(); in GreedyBySizeAssignment() local
/external/pytorch/test/functorch/
Dcommon_utils.py154 def get_bdim_choices(num_tensors): argument
172 num_tensors, _, running_mean=None, running_var=None, *args argument
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc74 void FlexModelTest::AddTensors(int num_tensors, const std::vector<int>& inputs, in AddTensors()

123