Searched refs:rank_constant (Results 1 – 1 of 1) sorted by relevance
80 Tensor rank_constant(DT_INT32, TensorShape({})); in Compile() local81 rank_constant.scalar<int32>()() = rank; in Compile()83 ctx->SetConstantOutput(0, rank_constant); in Compile()