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()
248 return element_shape_; in ElemShape()
254 Status s = element_shape_.MergeWith(candidate, &new_element_shape_); in SetElemShape()
258 element_shape_ = new_element_shape_; in SetElemShape()
407 PartialTensorShape element_shape_ TF_GUARDED_BY(mu_);
471 if (!element_shape_.IsCompatibleWith(value_t->shape())) { in LockedWriteOrAggregate()
478 element_shape_.DebugString(), " (consider setting infer_shape=False)."); in LockedWriteOrAggregate()
479 } else if (identical_element_shapes_ && !element_shape_.IsFullyDefined()) { in LockedWriteOrAggregate()
480 element_shape_ = PartialTensorShape(value_t->shape().dim_sizes()); in LockedWriteOrAggregate()
568 } 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
594 OP_REQUIRES_OK(context, context->GetAttr("element_shape", &element_shape_)); in TensorArrayPackOrGatherOp()
612 OP_REQUIRES_OK(ctx, tensor_array->SetElemShape(element_shape_)); in Compute()
638 OP_REQUIRES(ctx, element_shape_.IsFullyDefined(), in Compute()
641 element_shape_.DebugString(), in Compute()
646 element_shape_.AsTensorShape(&empty_shape); in Compute()
661 ctx, element_shape_.IsCompatibleWith(value_0_t->shape()), in Compute()
663 element_shape_.DebugString(), in Compute()
[all …]
|
| D | list_kernels.h | 285 OP_REQUIRES_OK(c, c->GetAttr("element_shape", &element_shape_)); in TensorListConcat()
291 if (!element_shape_.unknown_rank()) { in Compute()
293 gtl::ArraySlice<int64>(element_shape_.dim_sizes()).subspan(1)); in Compute()
469 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::__anonf33b5b270111::TensorArrayOp
|