Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtraining_ops.cc56 .TypeConstraint("T", kFloatAndComplexTypes),
114 .TypeConstraint("T", kFloatAndComplexTypes),
230 .TypeConstraint("T", kFloatAndComplexTypes),
278 Name("ResourceApplyAdagrad").TypeConstraint("T", kFloatAndComplexTypes),
333 Name("ResourceApplyAdagradV2").TypeConstraint("T", kFloatAndComplexTypes),
389 .TypeConstraint("T", kFloatAndComplexTypes),
562 Name("ResourceApplyAdam").TypeConstraint("T", kFloatAndComplexTypes),
736 Name("ResourceApplyRMSProp").TypeConstraint("T", kFloatAndComplexTypes),
747 .TypeConstraint("T", kFloatAndComplexTypes),
972 Name("ResourceApplyAdadelta").TypeConstraint("T", kFloatAndComplexTypes),
H A Dcholesky_op.cc34 REGISTER_XLA_OP(Name("Cholesky").TypeConstraint("T", kFloatAndComplexTypes),
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_op_registry.h50 constexpr std::array<DataType, 6> kFloatAndComplexTypes = { variable