Searched refs:multiples_shape (Results 1 – 2 of 2) sorted by relevance
45 const TensorShape multiples_shape = ctx->InputShape("multiples"); in Compile() local48 ctx, TensorShapeUtils::IsVector(multiples_shape), in Compile()50 multiples_shape.DebugString())); in Compile()51 OP_REQUIRES(ctx, input_shape.dims() == multiples_shape.num_elements(), in Compile()55 multiples_shape.dim_size(0))); in Compile()
3696 multiples_shape = op.inputs[1].get_shape().with_rank(1)3697 input_shape = op.inputs[0].get_shape().with_rank(multiples_shape[0])