Searched refs:component_shape (Results 1 – 6 of 6) sorted by relevance
533 TensorShape component_shape({dataset()->batch_size_}); in EnsureOutputAllocated() local534 component_shape.AppendShape(return_values->at(i).shape()); in EnsureOutputAllocated()539 component_shape); in EnsureOutputAllocated()578 TensorShape component_shape(result->output[i].shape()); in ProcessResult() local579 component_shape.set_dim(0, result->num_elements); in ProcessResult()583 component_shape); in ProcessResult()691 TensorShape component_shape(t.shape()); in ReadBatchResult() local692 component_shape.set_dim(0, dataset()->batch_size_); in ReadBatchResult()695 Tensor new_t(ctx->allocator(attr), t.dtype(), component_shape); in ReadBatchResult()
565 TensorShape component_shape( in GetBatch() local567 component_shape.set_dim(0, batches_[next_output_].error_index); in GetBatch()572 batches_[next_output_].outputs[i].dtype(), component_shape); in GetBatch()954 TensorShape component_shape({static_cast<uint32>(batch_size)}); in AllocateOutput() local955 component_shape.AppendShape(map_fn_outputs.at(i).shape()); in AllocateOutput()960 component_shape); in AllocateOutput()
61 for component_shape in self._tuple_shapes:63 component_message.CopyFrom(component_shape.message)
329 TensorShape component_shape({}); in GetNextInternal() local331 component_shape.AddDim(batch_component_shape.dim_size(i)); in GetNextInternal()335 if (batch_elements[i][component_index].shape() == component_shape) { in GetNextInternal()
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()
2328 Shape component_shape = in MakeElementGenerator() local2332 PrimitiveType component_element_type = component_shape.element_type(); in MakeElementGenerator()