Searched defs:GetTensorBatchDimIndex (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/util/
H A Dtensor_format.h130 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { function in namespace:tensorflow
415 dim_sizes[GetTensorBatchDimIndex(dims, format)] = N;

Completed in 103 milliseconds