Searched defs:diag_shape (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dmatrix_set_diag_op.cc29 const TensorShape diag_shape = context->InputShape(1); variable
50 OP_REQUIRES(context, expected_diag_shape == diag_shape,
55 " and diagonal shape: ", diag_shape.DebugString()));
/external/tensorflow/tensorflow/core/kernels/
H A Dmatrix_set_diag_op.cc52 const TensorShape& diag_shape = diag.shape(); variable
68 OP_REQUIRES(context, expected_diag_shape == diag_shape,
73 " and diagonal shape: ", diag_shape.DebugString()));

Completed in 183 milliseconds