Searched refs:dense_attr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 320 auto dense_attr = attr.dyn_cast<mlir::DenseElementsAttr>(); in ConvertToTensorProto() local
321 if (!dense_attr) return errors::InvalidArgument("Unsupported elements attr"); in ConvertToTensorProto()
325 ConvertElementsAttr<float>(dense_attr, output->mutable_float_val()); in ConvertToTensorProto()
328 ConvertHalfElementsAttr(dense_attr, output->mutable_half_val()); in ConvertToTensorProto()
331 ConvertElementsAttr(dense_attr, output->mutable_double_val()); in ConvertToTensorProto()
340 ConvertIntElementsAttr(dense_attr.cast<DenseIntElementsAttr>(), 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()
[all …]
|
| /external/llvm-project/mlir/test/IR/ |
| D | pretty-attributes.mlir | 9 "test.dense_attr"() {foo.dense_attr = dense<[1, 2, 3]> : tensor<3xi32>} : () -> ()
12 "test.non_elided_dense_attr"() {foo.dense_attr = dense<[1, 2]> : tensor<2xi32>} : () -> ()
21 "test.dense_splat"() {foo.dense_attr = dense<1> : tensor<3xi32>} : () -> ()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.cc | 418 auto dense_attr = input_defining_op.value().dyn_cast<DenseElementsAttr>(); in getUnquantizedBias() local
420 if (dense_attr) { in getUnquantizedBias()
422 rewriter.create<tosa::ConstOp>(op->getLoc(), output_type, dense_attr); in getUnquantizedBias()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 318 AttrValue dense_attr; in AsGraphDefInternal() local
319 b->BuildAttrValue(dense_types_, &dense_attr); in AsGraphDefInternal()
320 attrs.emplace_back("Tdense", dense_attr); in AsGraphDefInternal()
|