| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| 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
|
| D | binary_ops.cc | 164 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::__anon07d9859b0111::ConvertBinary
197 static constexpr std::array<DataType, 1> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::__anon07d9859b0111::ConvertBooleanBinary
|
| D | log_softmax.cc | 31 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertLogSoftmax
|
| 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
|
| D | like_ops.cc | 34 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertLikeOps
|
| D | data_format_vec_permute.cc | 52 static constexpr std::array<DataType, 1> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertDataFormatVecPermute
|
| D | tile.cc | 31 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertTile
|
| D | fill_ops.cc | 34 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ConvertFillBase
|
| D | einsum.cc | 670 static constexpr std::array<DataType, 3> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::__anonf1f03c760111::ConvertEinsum
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | op_converter_test.cc | 43 static constexpr std::array<DataType, 2> AllowedDataTypes() { in AllowedDataTypes() function in tensorflow::tensorrt::convert::ExampleOpConverter
|
| D | op_converter.h | 94 static constexpr std::array<DataType, 2> AllowedDataTypes() { in AllowedDataTypes() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| 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
|