Searched refs:component_shape (Results 1 – 8 of 8) sorted by relevance
61 for component_shape in self._tuple_shapes:63 component_message.CopyFrom(component_shape.message)
452 PartialTensorShape component_shape; in Shape() local453 TF_RETURN_IF_ERROR(unwrap(component.get())->Shape(&component_shape)); in Shape()454 if (!first_shape.IsIdenticalTo(component_shape)) { in Shape()460 component_shape.DebugString())); in Shape()
300 TensorShape component_shape({}); in GetNextInternal() local302 component_shape.AddDim(batch_component_shape.dim_size(i)); in GetNextInternal()305 &batch_component, &component_shape](int index) { in GetNextInternal()308 component_shape) { in GetNextInternal()
731 TensorShape component_shape(t.shape()); in ReadBatch() local732 component_shape.set_dim(0, batch_size); in ReadBatch()735 Tensor new_t(ctx->allocator(attr), t.dtype(), component_shape); in ReadBatch()827 TensorShape component_shape((*batch)[i].shape()); in ProcessBatch() local828 component_shape.set_dim(0, num_elements); in ProcessBatch()832 component_shape); in ProcessBatch()
496 TensorShape component_shape({batch_size}); in GetNextInternal() local499 component_shape.AppendShape(remaining_shape); in GetNextInternal()502 component_shape); in GetNextInternal()
508 TensorShape component_shape({dataset()->batch_size_}); in EnsureOutputAllocated() local509 component_shape.AppendShape(return_values->at(i).shape()); in EnsureOutputAllocated()514 component_shape); in EnsureOutputAllocated()
162 TensorShape component_shape(ready_tuples[0][i].shape()); in TryInsertMany() local163 component_shape.InsertDim(0, insertion_size); in TryInsertMany()164 Tensor component(ready_tuples[0][i].dtype(), component_shape); in TryInsertMany()
2372 Shape component_shape = in MakeElementGenerator() local2376 PrimitiveType component_element_type = component_shape.element_type(); in MakeElementGenerator()