Lines Matching refs:multiples
45 const TensorShape multiples_shape = ctx->InputShape("multiples");
49 errors::InvalidArgument("Expected multiples to be 1-D, but got shape ",
53 "Expected multiples argument to be a vector of length ",
58 // If input is a scalar then multiples has 0 elements and this is
65 std::vector<int64> multiples;
66 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("multiples", &multiples));
69 OP_REQUIRES(ctx, multiples[i] >= 0,
70 errors::InvalidArgument("Expected multiples[", i,
72 output_dims[i] = input_shape.dim_size(i) * multiples[i];
75 // If all multiples are 1, than the input is the same as the output.
76 if (absl::c_all_of(multiples,
84 int64 multiple = multiples[i];
114 REGISTER_XLA_OP(Name("Tile").CompileTimeConstantInput("multiples"), TileOp);
