Searched refs:kFloatTypes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tridiagonal_ops.cc | 71 REGISTER_XLA_OP(Name("TridiagonalSolve").TypeConstraint("T", kFloatTypes),
74 REGISTER_XLA_OP(Name("TridiagonalMatMul").TypeConstraint("T", kFloatTypes),
|
| D | xla_svd_op.cc | 97 REGISTER_XLA_OP(Name("XlaSvd").TypeConstraint("T", kFloatTypes), XlaSvdOp);
98 REGISTER_XLA_OP(Name("Svd").TypeConstraint("T", kFloatTypes), SvdOp);
|
| D | matrix_inverse_op.cc | 64 REGISTER_XLA_OP(Name("MatrixInverse").TypeConstraint("T", kFloatTypes),
|
| D | matrix_solve_op.cc | 72 REGISTER_XLA_OP(Name("MatrixSolve").TypeConstraint("T", kFloatTypes),
|
| D | training_ops.cc | 171 REGISTER_XLA_OP(Name("ResourceApplyMomentum").TypeConstraint("T", kFloatTypes),
465 REGISTER_XLA_OP(Name("ResourceApplyAdagradDA").TypeConstraint("T", kFloatTypes),
633 REGISTER_XLA_OP(Name("ResourceApplyAdaMax").TypeConstraint("T", kFloatTypes),
883 REGISTER_XLA_OP(Name("ResourceApplyFtrl").TypeConstraint("T", kFloatTypes),
905 REGISTER_XLA_OP(Name("ResourceApplyFtrlV2").TypeConstraint("T", kFloatTypes),
1054 REGISTER_XLA_OP(Name("ResourceApplyAddSign").TypeConstraint("T", kFloatTypes),
1074 REGISTER_XLA_OP(Name("ResourceApplyPowerSign").TypeConstraint("T", kFloatTypes),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 48 constexpr std::array<DataType, 4> kFloatTypes = { variable
|