- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 91 for foo_attr (0.11 sec)
-
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.td
"getElementTypeOrSelf($_self.cast<ElementsAttr>().getType()).isa<FloatType>()">, "float constant tensor">; // Checks if the boolean value is false. def IsFalseBoolAttr : AttrConstraint< CPred<"!$_self.cast<BoolAttr>().getValue()">>; // Checks if the value has only one user. def HasOneUse : Constraint<CPred<"$0.hasOneUse()">>; // Gets the type of a value. def GetValueType : NativeCodeCall<"$0.getType()">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 04:55:44 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
// This pass extracts a CPU computation cluster with `_xla_outside_compilation` // annotation from the head or tail of a TPU cluster. namespace { using mlir::Block; using mlir::BlockArgument; using mlir::BoolAttr; using mlir::ModuleOp; using mlir::OpBuilder; using mlir::Operation; using mlir::OperationPass; using mlir::Region; using mlir::StringAttr; using mlir::Type; using mlir::Value; using mlir::WalkResult;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
getOperation().walk([&](IslandOp island_op) { auto parent_func = island_op->getParentOfType<func::FuncOp>(); auto skip_island_outlining = parent_func->getAttrOfType<BoolAttr>(mlir::TF::kSkipIslandOutlining); if (skip_island_outlining && skip_island_outlining.getValue()) { // Island was marked to be skipped. return WalkResult::advance(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
#include "tensorflow/core/lib/monitoring/gauge.h" namespace tensorflow { namespace tf2xla { namespace internal { namespace { using mlir::Block; using mlir::BoolAttr; using mlir::Dialect; using mlir::LogicalResult; using mlir::MLIRContext; using mlir::ModuleOp; using mlir::Operation; using mlir::OperationName; using mlir::OperationPass; using mlir::Pattern;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
tensorflow/compiler/jit/node_matchers.h
impl::NodeMatcherProperties Attr(std::pair<string, AttrValue> attrs); impl::NodeMatcherProperties Attr(string name); std::pair<string, AttrValue> AttrLiteralHelper( const std::pair<string, bool>& bool_attr); std::pair<string, AttrValue> AttrLiteralHelper( const std::pair<string, absl::Span<const int>>& int_list_attr); std::pair<string, AttrValue> AttrLiteralHelper(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
/*fused_activation_function=*/rewriter.getStringAttr("NONE"), /*weights_format=*/rewriter.getStringAttr("DEFAULT"), /*keep_num_dims=*/rewriter.getBoolAttr(true), /*asymmetric_quantize_inputs=*/mlir::BoolAttr()); rewriter.replaceOp(bmm_op, {fc_op.getResult(0)}); return success(); }; }; // Converts batch_matmul operation with a ones tensor to a reduce_sum. struct ConvertBatchMatMulOpToReduceSum
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
LogicalResult PartitionResourceReadsWrites( tf_device::ClusterFuncOp cluster_func) { bool use_spmd = false; if (auto use_spmd_attr = cluster_func->getAttrOfType<BoolAttr>(kUseSpmdAttr)) use_spmd = use_spmd_attr.getValue(); if (!use_spmd) return success(); auto num_cores_per_replica_attr = cluster_func->getAttrOfType<IntegerAttr>(kNumCoresPerReplicaAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
// BatchNorm op patterns. //===----------------------------------------------------------------------===// def FalseBoolAttr : AttrConstraint<CPred<"!$_self.cast<BoolAttr>().getValue()">>; def TrueBoolAttr : AttrConstraint<CPred<"$_self.cast<BoolAttr>().getValue()">>; def CastValueToI64: NativeCodeCall< "CastValueToI64($0.getLoc(), $1, &$_builder)">; def CastValueToElementType: NativeCodeCall<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
mlir::DenseIntElementsAttr::get(static_shape_type, static_shape); return rewriter.create<TF::ConstOp>(loc, static_shape_attr).getOutput(); } // If the shape is not static, create a new ShapeOp. BoolAttr false_attr = rewriter.getBoolAttr(false); return rewriter .create<TF::ShapeOp>(loc, input, /*use_32bit=*/false_attr) .getOutput(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
(TFL_BatchMatMulOp:$bmm_tmp_output (TFL_ReshapeOp:$initial_shape_change $input, (Arith_ConstantOp $s0)), $rhs, $adj_x, $adj_y, $bool_attr), (Arith_ConstantOp $s1)), (TFL_BatchMatMulOp $input, $rhs, $adj_x, $adj_y, $bool_attr), [(HasRank<2> $rhs), (HasRank<2> $initial_shape_change), (BroadcastDimsProductEqual<0> $input, $initial_shape_change),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0)