OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:expected_diag_shape
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
matrix_set_diag_op.cc
48
TensorShape
expected_diag_shape
= batch_shape;
variable
49
expected_diag_shape
.AddDim(min_dim);
50
OP_REQUIRES(context,
expected_diag_shape
== diag_shape,
/external/tensorflow/tensorflow/core/kernels/
matrix_set_diag_op.cc
65
TensorShape
expected_diag_shape
= input_shape;
variable
66
expected_diag_shape
.RemoveLastDims(2);
67
expected_diag_shape
.AddDim(min_dim);
68
OP_REQUIRES(context,
expected_diag_shape
== diag_shape,
Completed in 68 milliseconds