Searched refs:kFloatAndComplexTypes (Results 1 – 2 of 2) 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),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 50 constexpr std::array<DataType, 6> kFloatAndComplexTypes = { variable
|