Searched defs:sum_shape (Results 1 – 3 of 3) sorted by relevance
40 TensorShape sum_shape; in Compile() local
258 TensorShape sum_shape = grad_val->shape(); in AddToAccumGradFunction() local
188 auto sum_shape = builder.GetShape(sum).ConsumeValueOrDie(); in XLA_TEST_P() local