Searched refs:kFloatTypes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 56 Name("ResourceApplyGradientDescent").TypeConstraint("T", kFloatTypes),
114 .TypeConstraint("T", kFloatTypes),
171 REGISTER_XLA_OP(Name("ResourceApplyMomentum").TypeConstraint("T", kFloatTypes),
230 Name("ResourceApplyKerasMomentum").TypeConstraint("T", kFloatTypes),
270 REGISTER_XLA_OP(Name("ResourceApplyAdagrad").TypeConstraint("T", kFloatTypes),
326 Name("ResourceApplyProximalAdagrad").TypeConstraint("T", kFloatTypes),
402 REGISTER_XLA_OP(Name("ResourceApplyAdagradDA").TypeConstraint("T", kFloatTypes),
487 REGISTER_XLA_OP(Name("ResourceApplyAdam").TypeConstraint("T", kFloatTypes),
558 REGISTER_XLA_OP(Name("ResourceApplyAdaMax").TypeConstraint("T", kFloatTypes),
660 REGISTER_XLA_OP(Name("ResourceApplyRMSProp").TypeConstraint("T", kFloatTypes),
[all …]
|
| D | xla_self_adjoint_eig_op.cc | 60 REGISTER_XLA_OP(Name("XlaSelfAdjointEig").TypeConstraint("T", kFloatTypes),
62 REGISTER_XLA_OP(Name("SelfAdjointEigV2").TypeConstraint("T", kFloatTypes),
|
| D | xla_svd_op.cc | 91 REGISTER_XLA_OP(Name("XlaSvd").TypeConstraint("T", kFloatTypes), XlaSvdOp);
92 REGISTER_XLA_OP(Name("Svd").TypeConstraint("T", kFloatTypes), SvdOp);
|
| D | cholesky_op.cc | 34 REGISTER_XLA_OP(Name("Cholesky").TypeConstraint("T", kFloatTypes), CholeskyOp);
|
| D | qr_op.cc | 44 REGISTER_XLA_OP(Name("Qr").TypeConstraint("T", kFloatTypes), QROp);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 48 constexpr std::array<DataType, 4> kFloatTypes = { variable
|