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