Searched refs:y_outer_dim (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ | ||
H A D | batch_dot.cc | 83 int y_outer_dim = transpose_y ? (ndims - 2) : (ndims - 1); local 85 dimensions.push_back(y_shape->dimensions(y_outer_dim)); |
Completed in 128 milliseconds