Home
last modified time | relevance | path

Searched refs:max_num_elements (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtensor_list.h73 max_num_elements(other.max_num_elements), in TensorList()
81 max_num_elements(rhs.max_num_elements), in TensorList()
90 max_num_elements = rhs.max_num_elements;
101 max_num_elements = rhs.max_num_elements;
123 int max_num_elements = -1; variable
134 out.max_num_elements = max_num_elements; in Copy()
Dtensor_list.cc47 core::PutVarint64(&metadata, static_cast<uint64>(max_num_elements)); in Encode()
92 max_num_elements = static_cast<int>(scratch); in Decode()
Dlist_kernels.cc153 empty.max_num_elements = max_num_elements_t.scalar<int32>()(); in Compute()
213 if (l->max_num_elements != -1) { in Compute()
215 c, l->tensors().size() < l->max_num_elements, in Compute()
218 " max_num_elements: ", l->max_num_elements)); in Compute()
405 output_list.max_num_elements = input_list->max_num_elements; in Compute()
688 to->max_num_elements = from.max_num_elements; in TensorListDeviceCopy()
/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
Dlist_ops_test.py50 def _testPushPop(self, max_num_elements): argument
54 max_num_elements=max_num_elements)
64 def testPushPop(self, max_num_elements): argument
65 self._testPushPop(max_num_elements)
70 def testPushPopGPU(self, max_num_elements): argument
72 self._testPushPop(max_num_elements)
77 element_dtype=dtypes.float32, element_shape=[], max_num_elements=1)
87 def testPopFromEmptyTensorListFails(self, max_num_elements): argument
91 max_num_elements=max_num_elements)
162 def _testStack(self, max_num_elements): argument
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_list_ops_test.py39 max_num_elements=20)
50 max_num_elements=10)
66 max_num_elements=10)
84 max_num_elements=20)
111 max_num_elements=2)
142 element_shape=[], element_dtype=dtypes.float32, max_num_elements=2)
171 element_dtype=dtypes.float32, element_shape=None, max_num_elements=2)
197 max_num_elements=2)
232 max_num_elements=2)
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DEmptyTensorList.pbtxt8 name: "max_num_elements"
37 name: "max_num_elements"
73 name: "max_num_elements"
/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dparse_tensor_op_fuzz.cc58 const int64_t max_num_elements = 1 << 18; in FuzzImpl() local
59 if (num_elements > max_num_elements) { in FuzzImpl()
/external/tensorflow/tensorflow/python/ops/
Dlist_ops.py49 max_num_elements=None, argument
51 if max_num_elements is None:
52 max_num_elements = -1
57 max_num_elements=max_num_elements,
Dwhile_v2.py281 max_num_elements=maximum_iterations)
1210 max_num_elements=self._maximum_iterations,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc220 int64_t max_num_elements; in Compile() local
223 1, &max_num_elements, xla::ValueInferenceMode::kUpperBound)); in Compile()
227 OP_REQUIRES(ctx, max_num_elements >= 0, in Compile()
251 element_shape, max_num_elements, in Compile()
255 ctx, element_shape, list_shape, max_num_elements); in Compile()
271 BuildUninitializedTensorList(ctx->builder(), max_num_elements, in Compile()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DEmptyTensorList.pbtxt8 name: "max_num_elements"
/external/icing/icing/file/
Dfile-backed-vector_test.cc719 int32_t max_num_elements = in TEST_F() local
728 ICING_ASSERT_OK(vector->Set(max_num_elements - 1, 'z')); in TEST_F()
729 ASSERT_THAT(vector->num_elements(), Eq(max_num_elements)); in TEST_F()
811 int32_t max_num_elements = in TEST_F() local
820 ICING_ASSERT_OK(vector->Set(max_num_elements - 3, 'z')); in TEST_F()
821 ASSERT_THAT(vector->num_elements(), Eq(max_num_elements - 2)); in TEST_F()
938 int32_t max_num_elements = (max_file_size - header_size) / element_type_size; in TEST_F() local
954 EXPECT_THAT(vector->Set(max_num_elements, 1), in TEST_F()
958 EXPECT_THAT(vector->Set(max_num_elements - 1, 1), IsOk()); in TEST_F()
960 int32_t start = max_num_elements - 5; in TEST_F()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2022-122.md11 tf.raw_ops.EmptyTensorList(element_shape=tf.ones(dtype=tf.int32, shape=[1, 0]), max_num_elements=tf…
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util_test.cc470 auto max_num_elements = in TEST() local
473 max_num_elements, DT_INT32); in TEST()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example.cc554 const size_t max_num_elements = max_num_features / stride_size; in FastParseExampleLite() local
559 values_shape.AddDim(max_num_elements); in FastParseExampleLite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc568 element_shape, list.max_num_elements(), list, list.element_dtype(), in HandleEmptyTensorListOp()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1390 const size_t max_num_elements = max_num_features / stride_size; in FastParseExample() local
1395 values_shape.AddDim(max_num_elements); in FastParseExample()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td234 TF_Int32Tensor:$max_num_elements
Dtf_ops_a_m.cc2359 if (!IsOfRankOrUnranked(op.max_num_elements(), 0)) { in verify()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt1393 …argspec: "args=[\'element_shape\', \'max_num_elements\', \'element_dtype\', \'name\'], varargs=Non…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt1393 …argspec: "args=[\'element_shape\', \'max_num_elements\', \'element_dtype\', \'name\'], varargs=Non…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go13851 func EmptyTensorList(scope *Scope, element_shape tf.Output, max_num_elements tf.Output, element_dty…
13859 element_shape, max_num_elements,
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt15275 name: "max_num_elements"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt21975 name: "max_num_elements"