Searched refs:sum_shape (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_conditional_accumulator.h258 TensorShape sum_shape = grad_val->shape(); variable
259 sum_shape.set_dim(0, sum_nnz);
262 ctx, ctx->allocate_persistent(dtype_, sum_shape, tensor_sum_persistent,
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc179 auto sum_shape = builder.GetShape(sum).ConsumeValueOrDie(); local
181 ShapeUtil::ElementsIn(*sum_shape));

Completed in 106 milliseconds