Lines Matching defs:input_sizes
554 const Tensor& input_sizes = context->input(0);
557 context, TensorShapeUtils::IsVector(input_sizes.shape()),
559 "Conv2DBackpropInput: input_sizes input must be 1-dim, not ",
560 input_sizes.dims()));
562 const int32* in_sizes_data = input_sizes.template flat<int32>().data();
563 for (int i = 0; i < input_sizes.NumElements(); ++i) {
566 " of input_sizes must be >= 0"));
608 .HostMemory("input_sizes"),
615 .HostMemory("input_sizes"),
