Home
last modified time | relevance | path

Searched refs:kNumV1Inputs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc242 context, context->num_inputs() >= kNumV1Inputs, in Compile()
265 if (context->num_inputs() > kNumV1Inputs) { in Compile()
327 static constexpr int kNumV1Inputs = 1; member in tensorflow::MatrixDiagOp
376 if (context->num_inputs() > kNumV1Inputs) { in Compile()
441 static constexpr int kNumV1Inputs = 1; member in tensorflow::MatrixDiagPartOp
486 if (context->num_inputs() > kNumV1Inputs) { in Compile()
534 static constexpr int kNumV1Inputs = 2; member in tensorflow::MatrixSetDiagOp
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_diag_op.cc68 if (context->num_inputs() > kNumV1Inputs) { in Compute()
146 static constexpr int kNumV1Inputs = 1; member in tensorflow::MatrixDiagPartOp
174 if (context->num_inputs() > kNumV1Inputs) { in Compute()
265 static constexpr int kNumV1Inputs = 1; member in tensorflow::MatrixDiagOp
Dmatrix_set_diag_op.cc65 if (context->num_inputs() > kNumV1Inputs) { in Compute()
172 static constexpr int kNumV1Inputs = 2; member in tensorflow::MatrixSetDiagOp