Searched refs:list_shape (Results 1 – 3 of 3) sorted by relevance
59 Status GetTensorListBufferShape(const xla::Shape& list_shape, in GetTensorListBufferShape() argument61 TF_RET_CHECK(list_shape.IsTuple()); in GetTensorListBufferShape()63 xla::ShapeUtil::GetTupleElementShape(list_shape, 0), buffer_shape)); in GetTensorListBufferShape()68 TensorShape list_shape; in IsTensorListInitialized() local69 TF_RETURN_IF_ERROR(GetTensorListBufferShape(op, &list_shape)); in IsTensorListInitialized()70 *is_initialized = !(list_shape.dims() == 2 && list_shape.dim_size(1) == 0); in IsTensorListInitialized()
66 TensorShape list_shape; in CreateZerosList() local67 list_shape.AddDim(leading_dim); in CreateZerosList()81 list_shape.AppendShape(element_shape); in CreateZerosList()91 list_shape.AddDim(0); in CreateZerosList()94 list_shape.dim_sizes()); in CreateZerosList()
49 Status GetTensorListBufferShape(const xla::Shape& list_shape,