- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for type_attr (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
return false; } } return true; } // Returns the tensor type created from the `shape_attr` and `type_attr` // attributes. Type GetType(Attribute shape_attr, Attribute type_attr) { auto shape = mlir::cast<tf_type::ShapeAttr>(shape_attr); auto type = mlir::cast<TypeAttr>(type_attr); if (shape.hasRank()) return tensorflow::GetTypeFromTFTensorShape(shape.getShape(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
type_attr = attr_value_pb2.AttrValue(type=types_pb2.DT_QINT8) if quantize: self.assertTrue( self._contains_op(output_graphdef, 'Const', 'dtype', type_attr) ) else: self.assertFalse( self._contains_op(output_graphdef, 'Const', 'dtype', type_attr) ) @parameterized.named_parameters(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
auto new_filter_constant_value_attr = DenseIntElementsAttr::get(new_filter_value_attr_type, new_filter_values); return rewriter.create<TFL::QConstOp>( filter_constant_op.getLoc(), /*output=*/TypeAttr::get(new_filter_result_type), /*value=*/new_filter_constant_value_attr); } // Creates the `tfl.qconst` for filter. If `rhs_op` is a `stablehlo.constant`,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
// determined while going through quantization passes. OptionalAttr<TypeAttr>:$input_to_input_intermediate, OptionalAttr<TypeAttr>:$input_to_forget_intermediate, OptionalAttr<TypeAttr>:$input_to_cell_intermediate, OptionalAttr<TypeAttr>:$input_to_output_intermediate, OptionalAttr<TypeAttr>:$effective_hidden_scale_intermediate ); let results = (outs AnyTensor:$output);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
return f.isExactlyValue(value); }); } // Returns true if `value` is compile-time constant and its splat value equals // to `raw_value`. template <typename T> bool IsConstantValueOf(mlir::TypedAttr value, T raw_value) { auto element_type = mlir::cast<ShapedType>(value.getType()).getElementType(); if (mlir::isa<FloatType>(element_type)) { return FloatValueEquals(value, raw_value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// Reshape for dynamic shaped operands. The result shape is // [lhs_batch_dimensions, lhs_out_dimensions, rhs_out_dimensions]. BoolAttr true_attr = rewriter.getBoolAttr(true); auto lhs_shape = rewriter.create<TF::ShapeOp>(loc, lhs, true_attr); auto rhs_shape = rewriter.create<TF::ShapeOp>(loc, rhs, true_attr); llvm::SmallVector<int64_t, 4> lhs_batch_and_out = Concat<int64_t>(lhs_dot_dimensions_info.batch_dimensions().AxesArray(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
src/cmd/internal/obj/x86/asm6.go
ctxt.Diag("offset too large in %s", p) } } v := int32(a.Offset) rel.Siz = 0 switch a.Type { case obj.TYPE_ADDR: if a.Name == obj.NAME_NONE { ctxt.Diag("unexpected TYPE_ADDR with NAME_NONE") } if a.Index == REG_TLS { ctxt.Diag("unexpected TYPE_ADDR with index==REG_TLS") } goto bad case obj.TYPE_REG: const regFirst = REG_AL const regLast = REG_Z31
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 15 15:44:14 UTC 2024 - 146.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
int64_t out = 0; for (int64_t outer = 0; outer < outer_size; ++outer) { for (auto op : operands) { auto typed_attr = op.cast<TypedAttr>(); const int64_t dim_size = typed_attr.getType().cast<RankedTensorType>().getDimSize(axis); const int64_t inner_size = dim_size * base_inner_size;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// types. But we need to wrap it up with ElementsAttr to construct // valid TensorFlow constants. auto typed_attr = mlir::cast<TypedAttr>(value); type = tensorflow::GetTypeFromTFTensorShape(/*shape=*/{}, typed_attr.getType()); return ConstOp::build(builder, result, DenseElementsAttr::get(type, value)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0)