HomeSort by relevance Sort by last modified time
    Searched refs:sum_shape (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
sparse_conditional_accumulator.h 258 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/
batch_normalization_test.cc 179 auto sum_shape = builder.GetShape(sum).ConsumeValueOrDie(); local
181 ShapeUtil::ElementsIn(*sum_shape));
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 118 milliseconds