Home
last modified time | relevance | path

Searched refs:InferBatchSize (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dtensor_utils.h50 static int64 InferBatchSize(
Dtensor_utils.cc85 int64 TensorUtils::InferBatchSize( in InferBatchSize() function in tensorflow::boosted_trees::utils::TensorUtils
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc214 const int64 batch_size = TensorUtils::InferBatchSize( in DoCompute()
426 const int64 batch_size = TensorUtils::InferBatchSize( in DoCompute()