Home
last modified time | relevance | path

Searched refs:kFloatTypes (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc56 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 …]
Dxla_self_adjoint_eig_op.cc60 REGISTER_XLA_OP(Name("XlaSelfAdjointEig").TypeConstraint("T", kFloatTypes),
62 REGISTER_XLA_OP(Name("SelfAdjointEigV2").TypeConstraint("T", kFloatTypes),
Dxla_svd_op.cc91 REGISTER_XLA_OP(Name("XlaSvd").TypeConstraint("T", kFloatTypes), XlaSvdOp);
92 REGISTER_XLA_OP(Name("Svd").TypeConstraint("T", kFloatTypes), SvdOp);
Dcholesky_op.cc34 REGISTER_XLA_OP(Name("Cholesky").TypeConstraint("T", kFloatTypes), CholeskyOp);
Dqr_op.cc44 REGISTER_XLA_OP(Name("Qr").TypeConstraint("T", kFloatTypes), QROp);
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.h48 constexpr std::array<DataType, 4> kFloatTypes = { variable