Searched refs:element_shape_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array.h | 153 element_shape_(element_shape), in TensorArray()
247 return element_shape_; in ElemShape()
253 Status s = element_shape_.MergeWith(candidate, &new_element_shape_); in SetElemShape()
257 element_shape_ = new_element_shape_; in SetElemShape()
405 PartialTensorShape element_shape_ TF_GUARDED_BY(mu_);
468 if (!element_shape_.IsCompatibleWith(value->shape())) { in LockedWriteOrAggregate()
475 element_shape_.DebugString(), " (consider setting infer_shape=False)."); in LockedWriteOrAggregate()
476 } else if (identical_element_shapes_ && !element_shape_.IsFullyDefined()) { in LockedWriteOrAggregate()
477 element_shape_ = PartialTensorShape(value->shape().dim_sizes()); in LockedWriteOrAggregate()
564 } else if (!element_shape_.IsFullyDefined()) { in LockedRead()
[all …]
|
| D | tensor_array_ops.cc | 165 OP_REQUIRES_OK(context, context->GetAttr("element_shape", &element_shape_)); in TensorArrayOp()
208 key, dtype_, *tensor_array_output_handle, size, element_shape_, in CreateTensorArray()
222 PartialTensorShape element_shape_; member in tensorflow::TensorArrayOp
593 OP_REQUIRES_OK(context, context->GetAttr("element_shape", &element_shape_)); in TensorArrayPackOrGatherOp()
611 OP_REQUIRES_OK(ctx, tensor_array->SetElemShape(element_shape_)); in Compute()
637 OP_REQUIRES(ctx, element_shape_.IsFullyDefined(), in Compute()
640 element_shape_.DebugString(), in Compute()
645 element_shape_.AsTensorShape(&empty_shape); in Compute()
659 ctx, element_shape_.IsCompatibleWith(value_0_t->shape()), in Compute()
661 element_shape_.DebugString(), in Compute()
[all …]
|
| D | list_kernels.h | 391 OP_REQUIRES_OK(c, c->GetAttr("element_shape", &element_shape_)); in TensorListConcat()
397 if (!element_shape_.unknown_rank()) { in Compute()
399 gtl::ArraySlice<int64_t>(element_shape_.dim_sizes()).subspan(1)); in Compute()
577 PartialTensorShape element_shape_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_array_ops.cc | 136 OP_REQUIRES_OK(ctx, ctx->GetAttr("element_shape", &element_shape_)); in TensorArrayOp()
160 if (element_shape_.IsFullyDefined()) { in Compile()
161 CHECK(element_shape_.AsTensorShape(&shape)); in Compile()
181 PartialTensorShape element_shape_; member in tensorflow::__anon649a86450111::TensorArrayOp
|