Home
last modified time | relevance | path

Searched refs:num_tensors (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_v2_ops.cc45 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in ValidateInputs() local
69 context, shape_and_slices.NumElements() == num_tensors, in ValidateInputs()
70 errors::InvalidArgument("Expected ", num_tensors, in ValidateInputs()
74 OP_REQUIRES(context, context->num_inputs() == num_tensors + kFixedInputs, in ValidateInputs()
76 "Got ", num_tensors, " tensor names but ", in ValidateInputs()
78 OP_REQUIRES(context, context->num_inputs() == num_tensors + kFixedInputs, in ValidateInputs()
80 "Expected a total of ", num_tensors + kFixedInputs, in ValidateInputs()
83 num_tensors, " names, but received ", context->num_inputs(), in ValidateInputs()
103 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in Compute() local
112 for (int i = 0; i < num_tensors; ++i) { in Compute()
Dscoped_allocator_ops_test.cc128 const string& name, int32 id, int32 num_tensors) { in BuildNodeDef() argument
133 .Attr("N", num_tensors) in BuildNodeDef()
137 .Input(FakeInput(num_tensors, dtype)) // list of tensors in BuildNodeDef()
145 int32 num_tensors) { in BuildNodeDefWithReshape() argument
151 .Attr("N", num_tensors) in BuildNodeDefWithReshape()
155 .Input(FakeInput(num_tensors, dtype)) // list of tensors in BuildNodeDefWithReshape()
162 const string& name, int32 id, int32 num_tensors) { in MakeOp() argument
163 BuildNodeDefWithReshape(shape, dtype, reshape, name, id, num_tensors); in MakeOp()
262 const string& name, int32 id, int32 num_tensors, in BuildNodeDef() argument
267 .Attr("N", num_tensors) in BuildNodeDef()
[all …]
Dtensor_array_ops.cc1257 int32 num_tensors = static_cast<int32>(tensor_lengths->NumElements()); in Compute() local
1260 cumulative_lengths.reserve(num_tensors); in Compute()
1262 for (int i = 0; i < num_tensors; ++i) { in Compute()
1286 std::vector<TensorShape> element_shapes(num_tensors, tensor_value->shape()); in Compute()
1287 for (int32 i = 0; i < num_tensors; ++i) { in Compute()
1292 if (dynamic_size && array_size < num_tensors) { in Compute()
1293 array_size = num_tensors; in Compute()
1297 ctx, array_size == num_tensors, in Compute()
1300 array_size, " vs. ", num_tensors, "), and the TensorArray is not ", in Compute()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor.h52 int num_tensors = tensor_list.size; in VectorOfTensors() local
54 all_data_.reserve(num_tensors); in VectorOfTensors()
55 all_shape_.reserve(num_tensors); in VectorOfTensors()
56 all_shape_ptr_.reserve(num_tensors); in VectorOfTensors()
58 for (int i = 0; i < num_tensors; ++i) { in VectorOfTensors()
67 for (int i = 0; i < num_tensors; ++i) { in VectorOfTensors()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt65 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.float32
66 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.uint8
67 DISABLED_concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.int32
68 DISABLED_concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.int64
69 concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.float32
70 concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.uint8
71 DISABLED_concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.int32
72 DISABLED_concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.int64
73 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=2,type=tf.float32
74 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=2,type=tf.uint8
[all …]
/external/tensorflow/tensorflow/lite/
Darena_planner.cc57 allocs_.resize(graph_info_->num_tensors()); in ResetAllocations()
71 std::vector<int> refcounts(graph_info_->num_tensors(), 0); in PlanAllocations()
74 std::vector<int> allocated(graph_info_->num_tensors(), false); in PlanAllocations()
75 std::vector<int> deallocated(graph_info_->num_tensors(), false); in PlanAllocations()
102 alloc_queue_.reserve(2 * graph_info_->num_tensors()); in PlanAllocations()
190 TF_LITE_ENSURE(context_, graph_info_->num_tensors() >= allocs_.size()); in ExecuteAllocations()
191 allocs_.resize(graph_info_->num_tensors()); in ExecuteAllocations()
196 for (int i = 0; i < graph_info_->num_tensors(); ++i) { in ExecuteAllocations()
Dgraph_info.h31 virtual size_t num_tensors() const = 0;
Dgraph_info.cc71 tensor_epochs_.resize(info_->num_tensors(), kEpochAlwaysReady); in Partition()
Dgraph_info_test.cc42 size_t num_tensors() const override { return tensors_.size(); } in num_tensors() function in tflite::__anon6e38e6c20111::SimpleTestGraph
Darena_planner_test.cc132 size_t num_tensors() const override { return graph_->tensors()->size(); } in num_tensors() function in tflite::__anonabd56e9b0111::TestGraphInfo
/external/tensorflow/tensorflow/compiler/tests/
Dconcat_ops_test.py83 num_tensors = np.random.randint(2, 10)
93 for i in np.arange(num_tensors):
115 for i in np.arange(num_tensors):
219 num_tensors = np.random.randint(1, 10)
222 concat_dim_sizes = np.random.randint(1, 5, size=num_tensors)
306 num_tensors = 1001
307 for i in np.arange(num_tensors):
322 for i in np.arange(num_tensors):
/external/tensorflow/tensorflow/python/debug/lib/
Dsource_utils_test.py295 _, is_tf_py_library, num_nodes, num_tensors, num_dumps, first_line = (
299 self.assertEqual(14, num_tensors)
321 _, is_tf_py_library, num_nodes, num_tensors, num_dumps, _ = (
325 self.assertEqual(2, num_tensors)
336 (file_path, is_tf_py_library, num_nodes, num_tensors, num_dumps,
341 self.assertEqual(14, num_tensors)
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc70 void FlexModelTest::AddTensors(int num_tensors, const std::vector<int>& inputs, in AddTensors() argument
73 interpreter_->AddTensors(num_tensors); in AddTensors()
74 for (int i = 0; i < num_tensors; ++i) { in AddTensors()
Dtest_util.h88 void AddTensors(int num_tensors, const std::vector<int>& inputs,
/external/tensorflow/tensorflow/python/kernel_tests/
Dconcat_op_test.py99 num_tensors = np.random.randint(2, 10)
109 for i in np.arange(num_tensors):
130 for i in np.arange(num_tensors):
268 num_tensors = np.random.randint(12, 20)
271 concat_dim_sizes = np.random.randint(1, 5, size=num_tensors)
518 num_tensors = 5000
520 num_tensors = 500
521 for i in np.arange(num_tensors):
535 for i in np.arange(num_tensors):
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dexternal_optimizer.py266 num_tensors = len(tensors)
281 callback(*augmented_fetch_vals[num_tensors:])
283 return augmented_fetch_vals[:num_tensors]
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.h617 virtual size_t num_tensors() const = 0;
655 size_t num_tensors() const override;
684 size_t num_tensors() const override;
709 size_t num_tensors() const override;
Dmessage_wrappers.cc796 size_t InMemoryRunStepResponse::num_tensors() const { return tensors_.size(); } in num_tensors() function in tensorflow::InMemoryRunStepResponse
839 size_t OwnedProtoRunStepResponse::num_tensors() const { in num_tensors() function in tensorflow::OwnedProtoRunStepResponse
891 size_t NonOwnedProtoRunStepResponse::num_tensors() const { in num_tensors() function in tensorflow::NonOwnedProtoRunStepResponse
Dmessage_wrappers_test.cc150 ASSERT_EQ(2, response.num_tensors()); in CheckRunStepResponse()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc997 Py_ssize_t num_tensors = PySequence_Fast_GET_SIZE(tensors); in TFE_Py_TensorShapeSlice() local
998 int64_t num_tensors_int = static_cast<int64_t>(num_tensors); in TFE_Py_TensorShapeSlice()
1003 for (Py_ssize_t i = 0; i < num_tensors; ++i) { in TFE_Py_TensorShapeSlice()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.h72 const string& key_prefix, int64 src_incarnation, int64 num_tensors,
Dprocess_function_library_runtime.cc131 const string& key_prefix, int64 src_incarnation, int64 num_tensors, in ReceiveTensorsAsync() argument
136 for (int64 i = 0; i < num_tensors; ++i) { in ReceiveTensorsAsync()
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py2036 num_tensors = self.node_def.attr['N'].i
2037 inputs = [inputs[:num_tensors]] + inputs[num_tensors:]
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli.py1222 for (file_path, _, num_nodes, num_tensors, num_dumps,
1236 str(num_tensors) + " " *
1237 (num_tensors_column_width - len(str(num_tensors))), color)
/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops.cc103 static size_t StringPaddingSize(size_t num_tensors) { in StringPaddingSize() argument
104 return strings::Printf("%zu", num_tensors - 1).size(); in StringPaddingSize()

12