tensorflow/core/kernels/sparse_cross_op.cc
// Calculate the batch size from either the shapes input or the dense input.
Calculate the batch size from either the shapes input or the dense input.
Calculate the batch size from either the shapes input or the dense input .
int64_t CalculateBatchSize(const OpInputList& shapes_list_in,
                           const OpInputList& dense_list_in) {
  if (shapes_list_in.size() > 0) {
    return shapes_list_in[0].vec<int64_t>()(0);
  }
  if (dense_list_in.size() > 0) {
    return dense_list_in[0].dim_size(0);
  }
  return 0;
}
int64_t CalculateBatchSize ( const OpInputList & shapes_list_in , const OpInputList & dense_list_in ) { if ( shapes_list_in . size ( ) > 0 ) { return shapes_list_in [ 0 ] . vec < int64_t > ( ) ( 0 ) ; } if ( dense_list_in . size ( ) > 0 ) { return dense_list_in [ 0 ] . dim_size ( 0 ) ; } return 0 ; }
Calculate the batch size from either the shapes input or the dense input.
Calculate the batch size from either the shapes input or the dense input .