Home
last modified time | relevance | path

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

12

/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/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/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
DDynamicTensorFixture.h423 const unsigned int num_tensors = num_functions + 1; in run() local
/external/ComputeLibrary/src/graph/nodes/
DStackLayerNode.cpp58 const unsigned int num_tensors = input_descriptors.size(); in compute_output_descriptor() 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/ComputeLibrary/tests/validation/reference/
DStackLayer.cpp53 const int num_tensors = in.size(); in stack_layer() local
/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/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/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc74 void FlexModelTest::AddTensors(int num_tensors, const std::vector<int>& inputs, in AddTensors()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc823 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
/external/tensorflow/tensorflow/python/debug/lib/
Ddumping_callback.py647 def _get_symbolic_tensor_ids(self, num_tensors): argument
/external/tensorflow/tensorflow/lite/
Dsimple_planner_test.cc136 size_t num_tensors() const override { return graph_->tensors()->size(); } in num_tensors() function in tflite::__anonf6c84ea30111::TestGraphInfo
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py844 def num_tensors(self): member in TrackableWeightHandler

12