| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.cc | 221 #define HANDLE_TF_TYPE(tftype, enumerant, name) tftype##Type, in TensorFlowDialect() argument
222 #define HANDLE_LAST_TF_TYPE(tftype, enumerant, name) tftype##Type in TensorFlowDialect() argument
360 #define HANDLE_TF_TYPE(tftype, enumerant, name) \ in parseType() argument
361 if (data == name) return tftype##Type::get(getContext()); in parseType()
364 #define HANDLE_CUSTOM_TF_TYPE(tftype, enumerant, name) in parseType() argument
376 #define HANDLE_TF_TYPE(tftype, enumerant, name) \ in printType() argument
377 if (auto derived_ty = ty.dyn_cast<tftype##Type>()) { \ in printType()
381 #define HANDLE_CUSTOM_TF_TYPE(tftype, enumerant, name) \ in printType() argument
382 if (auto derived_ty = ty.dyn_cast<tftype##Type>()) { \ in printType()
383 Print##tftype##Type(derived_ty, os); \ in printType()
|
| D | tf_types.cc | 92 #define HANDLE_TF_TYPE(tftype, enumerant, name) in classof() argument
93 #define HANDLE_TF_REF_TYPE(tftype, enumerant, name) tftype##Type, in classof() argument
94 #define HANDLE_LAST_TF_TYPE(tftype, enumerant, name) tftype##Type in classof() argument
142 #define HANDLE_TF_TYPE(tftype, enumerant, name) \ in get() argument
143 if (auto derived_ty = type.dyn_cast<tftype##Type>()) \ in get()
144 return tftype##RefType::get(ctx); in get()
146 #define HANDLE_TF_REF_TYPE(tftype, enumerant, name) in get() argument
175 #define HANDLE_TF_TYPE(tftype, enumerant, name) \ in RemoveRef() argument
176 if (isa<tftype##RefType>()) return tftype##Type::get(ctx); in RemoveRef()
178 #define HANDLE_TF_REF_TYPE(tftype, enumerant, name) in RemoveRef() argument
|
| D | tf_types.h | 154 #define HANDLE_TF_TYPE(tftype, enumerant, name) \ argument
155 class tftype##Type : public detail::TensorFlowTypeImpl<tftype##Type> { \
161 #define HANDLE_CUSTOM_TF_TYPE(tftype, enumerant, name) argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_type.cc | 81 #define HANDLE_TF_TYPE(tftype, enumerant, name) \ in ConvertDataType() argument
83 *type = builder.getType<mlir::TF::tftype##Type>(); \ in ConvertDataType()
140 #define HANDLE_TF_TYPE(tftype, enumerant, name) \ in ConvertScalarTypeToDataType() argument
141 if (type.isa<mlir::TF::tftype##Type>()) { \ in ConvertScalarTypeToDataType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | derived_size_attr.mlir | 13 …%dim:2 = tf_executor.island wraps "tf.Const"() {dtype = "tftype$DT_INT32", value = dense<0> : tens…
14 …%input:2 = tf_executor.island wraps "tf.Const"() {dtype = "tftype$DT_INT32", value = dense<1.0> : …
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | variable_ops_test.py | 48 def _initFetch(self, x, tftype, use_gpu=None): argument
50 p = state_ops.variable_op(x.shape, tftype)
62 tftype = _NP_TO_TF[dtype]
63 self.assertAllEqual(x, self._initFetch(x, tftype, use_gpu=False))
67 self.assertAllEqual(x, self._initFetch(x, tftype, use_gpu=True))
|