Searched refs:Tdims (Results 1 – 1 of 1) sorted by relevance
156 const Tensor& Tdims = context->input(0); in Compute() local157 OP_REQUIRES(context, IsLegacyVector(Tdims.shape()), in Compute()159 Tdims.shape().DebugString())); in Compute()164 auto dims = Tdims.flat<Index>(); in Compute()