| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | unary_ops.cc | 142 static constexpr std::array<DataType, 2> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertUnary
162 static constexpr std::array<DataType, 1> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertBooleanUnary
189 static constexpr std::array<DataType, 2> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertActivation
|
| H A D | binary_ops.cc | 164 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::__anon4e53edec0111::ConvertBinary
197 static constexpr std::array<DataType, 1> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::__anon4e53edec0111::ConvertBooleanBinary
|
| H A D | variable_ops.cc | 153 static constexpr std::array<DataType, 2> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertVariableV2
257 static constexpr std::array<DataType, 2> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertReadVariableOp
|
| H A D | log_softmax.cc | 31 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertLogSoftmax
|
| H A D | like_ops.cc | 34 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertLikeOps
|
| H A D | data_format_vec_permute.cc | 52 static constexpr std::array<DataType, 1> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertDataFormatVecPermute
|
| H A D | tile.cc | 31 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertTile
|
| H A D | fill_ops.cc | 34 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertFillBase
|
| H A D | einsum.cc | 670 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::__anonc98924470111::ConvertEinsum
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | op_converter_test.cc | 43 static constexpr std::array<DataType, 2> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ExampleOpConverter
|
| H A D | op_converter.h | 94 static constexpr std::array<DataType, 2> AllowedDataTypes() { in AllowedDataTypes() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | auto_mixed_precision.cc | 890 DataTypeSet AllowedDataTypes(const OpDef::AttrDef& attr_def) { in AllowedDataTypes() function
902 DataTypeSet AllowedDataTypes(const OpDef& op_def, const TypeAttrId& t_attr_id) { in AllowedDataTypes() function
|