Searched defs:second_tensor_dim (Results 1 – 1 of 1) sorted by relevance
992 size_t second_tensor_dim = ops[iter_ops]->inputs_tensor_info()[1].shape().size(); in CheckBroadcast() local1025 … size_t first_tensor_dim, size_t second_tensor_dim, bool broadcast_first_tensor) { in ApplyBroadcast()