Home | History | Annotate | Download | only in kernels

Lines Matching refs:sizes

37     const Tensor& sizes = context->input(1);
38     // Preliminary validation of sizes.
39     OP_REQUIRES(context, IsLegacyVector(sizes.shape()),
40                 errors::InvalidArgument("sizes input must be 1-D, not ",
41                                         sizes.shape().DebugString()));
48     switch (sizes.dtype()) {
50         OP_REQUIRES_OK(context, ValidateSizes<int32>(sizes, &product,
54         OP_REQUIRES_OK(context, ValidateSizes<int64>(sizes, &product,
60             DataTypeString(sizes.dtype())));
68                                   "input sizes are non-zero"));
94   Status ValidateSizes(const Tensor& sizes, int64* product, int* unknown_index,
98     const int64 num_dims = sizes.NumElements();
99     auto Svec = sizes.flat<Tshape>();