Searched refs:kFloatAndComplexTypes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 56 .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),
|
| D | cholesky_op.cc | 34 REGISTER_XLA_OP(Name("Cholesky").TypeConstraint("T", kFloatAndComplexTypes),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 50 constexpr std::array<DataType, 6> kFloatAndComplexTypes = { variable
|