Searched refs:num_in_shape (Results 1 – 1 of 1) sorted by relevance
127 const TensorShape num_in_shape = ctx->InputShape("num"); in Compile() local134 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(num_in_shape), in Compile()136 num_in_shape.DebugString())); in Compile()