- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for shape_value (0.44 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc
// Broadcast `multiplier` and `rhs` Value shape_value; if (!input_type.hasStaticShape()) { shape_value = getShapeValue(bn_op.getLoc(), bn_op.getOperand(), rewriter); } int64_t feature_dim = bn_op.getFeatureIndex(); auto broadcast_multiplier = broadcastToFeatureDim(bn_op.getLoc(), input_type, multiplier, shape_value, feature_dim, rewriter);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
namespace { Value GetDimValue(OpBuilder &builder, Location loc, Value shape_value, int32_t dim) { Type attribute_type = builder.getI64Type(); return builder.create<TF::StridedSliceOp>( loc, RankedTensorType::get( {}, mlir::cast<ShapedType>(shape_value.getType()).getElementType()), /*input=*/shape_value, /*begin=*/Create1DConstValue<int32_t>(builder, loc, {dim}),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
return success(); } // Parses an R1 value to `shape` if it is a TF::ConstOp output. Otherwise, // returns an error. LogicalResult GetConstShapeValue(Value shape_value, llvm::SmallVector<int64_t, 8>* shape) { auto shape_op = shape_value.getDefiningOp(); if (!shape_op) return failure(); auto shape_const_op = llvm::dyn_cast<TF::ConstOp>(shape_op); if (!shape_const_op) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
<< "output must have the same rank with the length of shape = " << output_rank; } } } DenseIntElementsAttr shape_value; if (matchPattern(shape, m_Constant(&shape_value))) { for (const auto shape_elem : shape_value) { if (shape_elem.getSExtValue() <= 0) { return op.emitOpError("all elements of shape must be > 0"); } }
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/lite/transforms/prepare_tf.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto reshape_sizes = DenseIntElementsAttr::get(reshape_type, new_shape); auto reshape_value = builder->create<TF::ConstOp>(input.getLoc(), reshape_sizes); return builder->create<TF::ReshapeOp>(input.getLoc(), reshape_result_type, input, reshape_value); } // Creates transpose op for shape to depth transform.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// %shape = [MS1, MS2] SmallVector<Value, 4> shape_values; shape_values.reserve(input_rank); for (int64_t i = 0; i < input_rank; ++i) { Value dim_size_value = rewriter.create<mlir::arith::MulIOp>( loc, out_dim_size[2 * i], out_dim_size[2 * i + 1]); shape_values.push_back(dim_size_value); } Value shape = rewriter.create<tensor::FromElementsOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
pkg/workloadapi/security/authorization.pb.go
Scope_WORKLOAD_SELECTOR Scope = 2 ) // Enum value maps for Scope. var ( Scope_name = map[int32]string{ 0: "GLOBAL", 1: "NAMESPACE", 2: "WORKLOAD_SELECTOR", } Scope_value = map[string]int32{ "GLOBAL": 0, "NAMESPACE": 1, "WORKLOAD_SELECTOR": 2, } ) func (x Scope) Enum() *Scope { p := new(Scope) *p = x return p }
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Fri May 31 02:27:10 UTC 2024 - 29.7K bytes - Viewed (0)