Searched refs:GetXlaConvTypes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_ops.cc | 73 REGISTER_XLA_OP(Name("Conv2D").TypeConstraint("T", GetXlaConvTypes()),
81 REGISTER_XLA_OP(Name("Conv3D").TypeConstraint("T", GetXlaConvTypes()),
90 Name("DepthwiseConv2dNative").TypeConstraint("T", GetXlaConvTypes()),
137 .TypeConstraint("T", GetXlaConvTypes()),
147 .TypeConstraint("T", GetXlaConvTypes()),
157 .TypeConstraint("T", GetXlaConvTypes()),
199 .TypeConstraint("T", GetXlaConvTypes()),
210 .TypeConstraint("T", GetXlaConvTypes()),
220 .TypeConstraint("T", GetXlaConvTypes()),
|
| D | conv_op_helpers.h | 40 std::vector<DataType> GetXlaConvTypes();
|
| D | extract_image_patches_op.cc | 184 Name("ExtractImagePatches").TypeConstraint("T", GetXlaConvTypes()),
|
| D | conv_op_helpers.cc | 156 std::vector<DataType> GetXlaConvTypes() { in GetXlaConvTypes() function
|