Searched refs:StrideOrInflate (Results 1 – 1 of 1) sorted by relevance
361 static Status StrideOrInflate(OpKernelContext* ctx, const Tensor& input, in StrideOrInflate() function487 StrideOrInflate<Device, T>(ctx, input_transposed, *labels, label_counts, in ReduceOperand()680 ctx, EinsumHelper::StrideOrInflate<Device, T>( in Compute()