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

  /external/tensorflow/tensorflow/core/kernels/
sparse_tensor_dense_add_op.cc 55 const auto a_shape_flat = a_shape->flat<Index>(); local
57 if (a_shape_flat(i) != b->dim_size(i)) {
61 a_shape_flat(i), " vs dense side ", b->dim_size(i));
sparse_add_op.cc 79 const auto a_shape_flat = a_shape->flat<int64>(); variable
82 OP_REQUIRES(ctx, a_shape_flat(i) == b_shape_flat(i),
84 "Operands' shapes do not match: got ", a_shape_flat(i),

Completed in 74 milliseconds