Lines Matching defs:input_sizes
221 const Tensor& input_sizes = context->input(0);
225 context, TensorShapeUtils::IsVector(input_sizes.shape()),
227 "Conv2DBackpropInput: input_sizes input must be 1-dim, not ",
228 input_sizes.dims()));
231 input_sizes.vec<int32>(), &input_shape));
336 const Tensor& input_sizes = context->input(0);
340 context, TensorShapeUtils::IsVector(input_sizes.shape()),
342 "Conv2DBackpropInput: input_sizes input must be 1-dim, not ",
343 input_sizes.dims()));
346 input_sizes.vec<int32>(), &input_shape));
654 const Tensor& input_sizes = context->input(0);
658 context, TensorShapeUtils::IsVector(input_sizes.shape()),
660 "Conv2DBackpropInput: input_sizes input must be 1-dim, not ",
661 input_sizes.dims()));
664 input_sizes.vec<int32>(), &input_shape));
1099 .HostMemory("input_sizes"),
1104 .HostMemory("input_sizes"),
