Searched defs:logits_sum_shape (Results 1 – 1 of 1) sorted by relevance
70 TensorShape logits_sum_shape = src->tensor_shape(); in is_supported_op() local138 TensorShape logits_sum_shape = src->tensor_shape(); in create_op() local