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