Searched refs:ConvertElementsAttr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 262 void ConvertElementsAttr(const mlir::DenseElementsAttr attr, in ConvertElementsAttr() function
325 ConvertElementsAttr<float>(dense_attr, output->mutable_float_val()); in ConvertToTensorProto()
331 ConvertElementsAttr(dense_attr, output->mutable_double_val()); in ConvertToTensorProto()
344 ConvertElementsAttr(dense_attr, output->mutable_uint32_val()); in ConvertToTensorProto()
347 ConvertElementsAttr(dense_attr, output->mutable_uint64_val()); in ConvertToTensorProto()
350 ConvertElementsAttr(dense_attr, output->mutable_int64_val()); in ConvertToTensorProto()
353 ConvertElementsAttr(dense_attr, output->mutable_bool_val()); in ConvertToTensorProto()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/utils/ |
| D | convert_op_folder.h | 28 mlir::ElementsAttr ConvertElementsAttr(const mlir::ElementsAttr& elements,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | convert_op_folder.cc | 27 mlir::ElementsAttr ConvertElementsAttr(const mlir::ElementsAttr& elements, in ConvertElementsAttr() function
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 56 "hlo::ConvertElementsAttr("
|
| D | legalize_tf.cc | 1016 return hlo::ConvertElementsAttr(slice_sizes, builder->getIntegerType(64)) in TFSliceSizes2HLOSliceSizes()
5267 hlo::ConvertElementsAttr(group_assignment, rewriter.getIntegerType(64)) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 543 return hlo::ConvertElementsAttr(elementsAttr, in fold()
|