Searched refs:size_input_idx (Results 1 – 1 of 1) sorted by relevance
31 int size_input_idx, DimensionHandle channel_dim) { in SetOutputToSizedImage() argument34 TF_RETURN_IF_ERROR(c->WithRank(c->input(size_input_idx), 1, &size)); in SetOutputToSizedImage()39 const Tensor* size_tensor = c->input_tensor(size_input_idx); in SetOutputToSizedImage()51 DataTypeString(size_tensor->dtype()), " for input #", size_input_idx, in SetOutputToSizedImage()