Home
last modified time | relevance | path

Searched refs:max_images_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsummary_image_op.cc38 max_images_ = static_cast<int32>(max_images_tmp); in SummaryImageOp()
148 const int N = std::min<int>(max_images_, batch_size); in AddImages()
157 if (max_images_ > 1) { in AddImages()
251 int32 max_images_; member in tensorflow::SummaryImageOp
Dsummary_kernels.cc224 max_images_ = static_cast<int32>(max_images_tmp); in WriteImageSummaryOp()
246 OP_REQUIRES_OK(ctx, s->WriteImage(step, *t, tag, max_images_, *bad_color)); in Compute()
250 int32 max_images_; member in tensorflow::WriteImageSummaryOp