| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | list_ops_test.py | 48 def _testPushPop(self, max_num_elements): argument
52 max_num_elements=max_num_elements)
59 def testPushPop(self, max_num_elements): argument
60 self._testPushPop(max_num_elements)
64 def testPushPopGPU(self, max_num_elements): argument
68 self._testPushPop(max_num_elements)
73 element_dtype=dtypes.float32, element_shape=[], max_num_elements=1)
83 def testPopFromEmptyTensorListFails(self, max_num_elements): argument
87 max_num_elements=max_num_elements)
145 def _testStack(self, max_num_elements): argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_list_ops_test.py | 40 max_num_elements=20)
51 max_num_elements=10)
67 max_num_elements=10)
85 max_num_elements=20)
112 max_num_elements=2)
143 element_shape=[], element_dtype=dtypes.float32, max_num_elements=2)
172 element_dtype=dtypes.float32, element_shape=None, max_num_elements=2)
198 max_num_elements=2)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.cc | 46 max_num_elements(other.max_num_elements) {} in TensorList()
67 core::PutVarint64(&metadata, static_cast<uint64>(max_num_elements)); in Encode()
79 to->max_num_elements = from.max_num_elements; in TensorListDeviceCopy()
138 max_num_elements = static_cast<int>(scratch); in Decode()
238 empty.max_num_elements = max_num_elements_t.scalar<int32>()(); in Compute()
294 if (l->max_num_elements != -1) { in Compute()
296 c, l->tensors.size() < l->max_num_elements, in Compute()
299 " max_num_elements: ", l->max_num_elements)); in Compute()
441 output_list.max_num_elements = input_list->max_num_elements; in Compute()
|
| D | list_kernels.h | 64 int max_num_elements = -1; member
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | parse_tensor_op_fuzz.cc | 56 const int64 max_num_elements = 1 << 20; in FuzzImpl() local
57 if (num_elements > max_num_elements) { in FuzzImpl()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | list_ops.py | 47 max_num_elements=None, argument
49 if max_num_elements is None:
50 max_num_elements = -1
55 max_num_elements=max_num_elements,
|
| D | while_v2.py | 768 max_num_elements=self._maximum_iterations)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 137 int64 max_num_elements; in Compile() local
138 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &max_num_elements)); in Compile()
140 ctx, max_num_elements >= 0, in Compile()
146 CreateZerosList(ctx, 0, max_num_elements, dtype_, &buffer)); in Compile()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1175 const size_t max_num_elements = max_num_features / stride_size; in FastParseExample() local
1180 values_shape.AddDim(max_num_elements); in FastParseExample()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 965 …argspec: "args=[\'element_shape\', \'max_num_elements\', \'element_dtype\', \'name\'], varargs=Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 965 …argspec: "args=[\'element_shape\', \'max_num_elements\', \'element_dtype\', \'name\'], varargs=Non…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 22467 func EmptyTensorList(scope *Scope, element_shape tf.Output, max_num_elements tf.Output, element_dty…
22475 element_shape, max_num_elements,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 10260 name: "max_num_elements"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 21975 name: "max_num_elements"
|
| D | ops_history.v1.pbtxt | 22346 name: "max_num_elements"
|