Searched refs:GetInputsToDeterminateBatchSize (Results 1 – 1 of 1) sorted by relevance
471 absl::Span<const OpInfo::TensorProperties> GetInputsToDeterminateBatchSize( in GetInputsToDeterminateBatchSize() function529 GetInputsToDeterminateBatchSize(node, all_input_properties); in OperationCanBeTranslatedToImplicitBatch()664 FindLeadingShape(GetInputsToDeterminateBatchSize(node, input_properties)); in GetClusterBatchSizeForNode()