Searched refs:InferBatchSize (Results 1 – 3 of 3) sorted by relevance
50 static int64 InferBatchSize(
85 int64 TensorUtils::InferBatchSize( in InferBatchSize() function in tensorflow::boosted_trees::utils::TensorUtils
214 const int64 batch_size = TensorUtils::InferBatchSize( in DoCompute()426 const int64 batch_size = TensorUtils::InferBatchSize( in DoCompute()