| /external/tensorflow/tensorflow/core/kernels/ |
| list_kernels.h | 55 PartialTensorShape element_shape; member in struct:tensorflow::TensorList
84 OP_REQUIRES(c, l->element_shape.IsFullyDefined(),
96 for (TensorShapeDim s : l->element_shape) {
109 c, l->element_shape.IsCompatibleWith(t.shape()),
112 l->element_shape.DebugString(),
143 PartialTensorShape element_shape; variable
144 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(1), &element_shape));
150 OP_REQUIRES(c, element_shape.IsCompatibleWith(output_shape),
152 "Specified a list with shape ", element_shape.DebugString(),
154 output_list.element_shape = element_shape
[all...] |
| list_kernels.cc | 44 element_shape(other.element_shape),
54 if (!element_shape.unknown_rank()) {
55 for (TensorShapeDim dim : element_shape) {
69 to->element_shape = from.element_shape;
153 PartialTensorShape element_shape; variable
154 OP_REQUIRES_OK(ctx, TensorShapeFromTensor(ctx->input(0), &element_shape));
155 empty.element_shape = element_shape;
356 PartialTensorShape element_shape; variable
[all...] |
| tensor_array.h | 140 int32 N, const PartialTensorShape& element_shape,
154 element_shape_(element_shape),
410 // shaped zero tensors on stack -- even if the initial element_shape
559 TensorShape element_shape; local
565 element_shape = tensors_[index].shape;
575 "element_shape property on the forward TensorArray, the proper "
579 element_shape_.AsTensorShape(&element_shape); // Always succeeds.
586 tensors_[i].shape = element_shape;
590 tensors_[index].shape = element_shape;
|
| barrier_ops.cc | 92 TensorShape element_shape = values.shape(); local
94 ctx, keys.NumElements() == 0 || element_shape.num_elements() > 0,
97 element_shape.DebugString()),
99 if (element_shape.dims() > 0) element_shape.RemoveDim(0);
133 InsertOneLocked<T>(ctx, keys, values, element_shape,
308 const Tensor& values, const TensorShape& element_shape,
372 values.dtype(), element_shape, &next_element, &allocated_element));
|
| tensor_array_ops.cc | 163 OP_REQUIRES_OK(context, context->GetAttr("element_shape", &element_shape_));
566 OP_REQUIRES_OK(context, context->GetAttr("element_shape", &element_shape_));
634 errors::InvalidArgument("TensorArray was passed element_shape ",
[all...] |
| fifo_queue.cc | 92 TensorShape element_shape(tuple[component].shape());
93 element_shape.RemoveDim(0);
96 tuple[component].dtype(), element_shape, out_tensor, &element_access));
|
| priority_queue.cc | 120 TensorShape element_shape(tuple[component].shape());
121 element_shape.RemoveDim(0);
124 tuple[component].dtype(), element_shape, out_tensor, &element_access));
|
| random_shuffle_queue_op.cc | 168 TensorShape element_shape(tuple[component].shape());
169 element_shape.RemoveDim(0);
172 tuple[component].dtype(), element_shape, out_tensor, &element_access));
|
| padding_fifo_queue.cc | 61 TensorShape element_shape(tuple[component].shape());
64 tuple[component].dtype(), element_shape, out_tensor, &element_access));
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| list_ops_test.py | 46 element_shape=scalar_shape())
59 element_shape=scalar_shape())
73 l = list_ops.tensor_list_from_tensor(t, element_shape=scalar_shape())
88 l = list_ops.tensor_list_from_tensor(t, element_shape=scalar_shape())
103 element_shape=-1)
113 l = list_ops.tensor_list_from_tensor(t, element_shape=scalar_shape())
148 l = list_ops.tensor_list_from_tensor(t, element_shape=[1])
160 element_shape=scalar_shape())
172 l = list_ops.tensor_list_from_tensor(c, element_shape=scalar_shape())
182 l = list_ops.tensor_list_from_tensor(c, element_shape=scalar_shape()
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| list_ops.cc | 24 .Input("element_shape: shape_type")
160 .Input("element_shape: shape_type")
171 shape_inference::ShapeHandle element_shape;
172 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &element_shape));
173 TF_RETURN_IF_ERROR(c->Merge(o, element_shape, &o));
175 0, std::vector<shape_inference::ShapeAndType>{{element_shape, t}});
181 .Output("element_shape: shape_type")
194 .Input("element_shape: shape_type")
218 shape_inference::ShapeHandle element_shape = c->UnknownShape();
222 element_shape = list_shape_type.shape
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| list_ops.py | 43 element_shape=gen_list_ops.tensor_list_element_shape(
51 element_shape=dtensor.shape[1:])
64 element_shape=gen_list_ops.tensor_list_element_shape(
|
| tensor_array_ops.py | 57 element_shape=None,
80 element_shape: (optional, default: None) A `TensorShape` object specifying
104 if handle is not None and element_shape is not None:
105 raise ValueError("Cannot provide both a handle and element_shape "
130 # shape is defined either by `element_shape` or the shape of the tensor
133 if element_shape is None:
138 self._element_shape = [tensor_shape.TensorShape(element_shape)]
154 element_shape=element_shape,
296 element_shape = self._element_shape[0
[all...] |
| rnn.py | 739 def _create_ta(name, element_shape, dtype):
742 element_shape=element_shape,
750 element_shape=(tensor_shape.TensorShape([const_batch_size])
758 element_shape=flat_input_i.shape[1:],
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| layout_util.cc | 114 for (auto& element_shape : *shape->mutable_tuple_shapes()) {
115 SetToDefaultLayout(&element_shape);
149 for (auto& element_shape : shape.tuple_shapes()) {
150 TF_RETURN_IF_ERROR(ValidateLayoutInShape(element_shape));
232 for (auto& element_shape : *shape->mutable_tuple_shapes()) {
233 ClearLayout(&element_shape);
|
| xla.proto | 421 Shape element_shape = 3;
427 // Maximum number of elements (with shape element_shape) to load.
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| copy_insertion_test.cc | 1470 const Shape element_shape = ShapeUtil::MakeShape(F32, {42}); local
1611 const Shape element_shape = ShapeUtil::MakeShape(F32, {42}); local
1730 const Shape element_shape = ShapeUtil::MakeShape(F32, {}); local
1749 const Shape element_shape = ShapeUtil::MakeShape(F32, {}); local
[all...] |
| /external/opencv/cv/src/ |
| cvmorph.cpp | 377 void CvMorphology::init_binary_element( CvMat* element, int element_shape, CvPoint anchor )
405 if( element_shape != RECT && element_shape != CROSS && element_shape != ELLIPSE )
412 element_shape = RECT;
414 if( element_shape == ELLIPSE )
426 if( element_shape == RECT || (element_shape == CROSS && i == anchor.y) )
428 else if( element_shape == CROSS )
[all...] |
| /external/tensorflow/tensorflow/core/kernels/data/ |
| padded_batch_dataset_op.cc | 251 const TensorShape& element_shape = variable
255 if (element_shape.dims() != padded_shape.dims()) {
260 " but got element with rank ", element_shape.dims());
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| test_utils.cc | 91 for (const Shape& element_shape : shape.tuple_shapes()) {
93 MakeFakeLiteralInternal(element_shape, engine));
|
| while_test.cc | 962 auto element_shape = ShapeUtil::MakeShape(F32, {2}); local
998 auto element_shape = ShapeUtil::MakeShape(F32, {2}); local
1024 auto element_shape = ShapeUtil::MakeShape(F32, {}); local
[all...] |
| local_client_execute_test.cc | 429 const Shape element_shape = ShapeUtil::MakeShape(F32, {2}); local
430 std::vector<Shape> element_shapes(kElementCount, element_shape);
476 const Shape element_shape = ShapeUtil::MakeShape(F32, {}); local
477 std::vector<Shape> element_shapes(kFanout, element_shape);
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| input.py | 112 element_shape=None,
129 `element_shape` must be defined.
130 element_shape: (Optional.) A `TensorShape` representing the shape of a
169 element_shape = input_tensor.shape[1:].merge_with(element_shape)
170 if not element_shape.is_fully_defined():
172 "or `element_shape` must be specified")
181 shapes=[element_shape],
252 element_shape=[],
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| embedding_ops.py | [all...] |
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| decoder.py | 228 element_shape=_shape(decoder.batch_size, s))
|
