- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 119 for getRand (0.25 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
mlir::dyn_cast_or_null<ShapedType>(operand.getType()); return result_type && operand_type && result_type.hasRank() && operand_type.hasRank() && result_type.getRank() == operand_type.getRank(); }); if (!is_valid_move) return; } // At this point we checked that we can safely move Transpose node before // `op`, and bypass all result transposes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_op_order.cc
if (passthrough_op->hasTrait<OpTrait::IsTerminator>()) return failure(); auto get_num_elements = [](RankedTensorType tensor) { int num_elements = 1; for (int i = 0; i < tensor.getRank(); ++i) { // Assume dynamic dim size as the dim size one. if (!tensor.isDynamicDim(i)) { num_elements *= tensor.getDimSize(i); } } return num_elements; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_arith_ops_folder.h
// Scalar identity is broadcastable to any operand shape, we only need to // check that operand has the same shape as a result. bool scalar_identity = identity_ty.hasRank() && identity_ty.getRank() == 0; if (scalar_identity) return operand_ty == result_ty; // If identity is not a scalar, we must verify that identity shape is // statically known to be broadcastable to the operand shape and the operand
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
CPred<"$0.getType().cast<ShapedType>().hasRank() && " "$0.getType().cast<ShapedType>().getRank() <= " # n>>; // Checks if the value has rank 'n'. class HasRank<int n> : Constraint< CPred<"$0.getType().cast<ShapedType>().hasRank() && " "$0.getType().cast<ShapedType>().getRank() == " # n>>; class FloatValueEquals<string val> : Constraint<CPred< "FloatValueEquals($0, " # val # ")">>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
staging/src/k8s.io/apiextensions-apiserver/test/integration/basic_test.go
} if e, a := noxuDefinition.Spec.Group+"/"+watchVersion, createdTypeMeta.GetAPIVersion(); e != a { t.Errorf("expected %v, got %v", e, a) } if e, a := noxuDefinition.Spec.Names.Kind, createdTypeMeta.GetKind(); e != a { t.Errorf("expected %v, got %v", e, a) } case <-time.After(5 * time.Second): t.Errorf("missing watch event") } } // Check get for all versions
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue Apr 23 11:35:33 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
BlockArgument arg = func.getArgument(index); auto inputType = mlir::dyn_cast<RankedTensorType>(arg.getType()); // Only rank 1 tensor is supported for now. if (!inputType || inputType.getRank() != 1) continue; auto shape = llvm::to_vector<4>(inputType.getShape()); llvm::SmallVector<int64_t, 4> bounds(shape.begin(), shape.end()); // Mark the dim as dynamic dim.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
if (!(tensor_type && tensor_type.hasRank())) { return op->emitError() << "cannot convert op with unranked or non-tensor input type " << tensor_type << "."; } int rank = tensor_type.getRank(); if (rank <= partition_dim) { return op->emitError() << "cannot partition " << first_operand_type << " (rank = " << rank << ") along dimension "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
hlo::convertElementsAttr(group_assignment, builder.getIntegerType(64))); if (replica_groups.getType().getRank() != 2) { return op->emitOpError() << "group_assignment should have rank 2, got " << replica_groups.getType().getRank(); } return success(); } ChannelHandleAttr ConvertChannel(OpBuilder& builder, int64_t channel_id,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.td
// Checks if the value has rank at most 'n'. class HasRankAtLeast<int n> : Constraint< CPred<"$0.getType().cast<ShapedType>().hasRank() && " "$0.getType().cast<ShapedType>().getRank() >= " # n>>; // Checks value is not produced by a TFL_Quant or // from TFL_Quant Op with same quant type. def NotFromQuantOpOrSameQuantType : Constraint< CPred<"tflite::NotFromQuantOpOrSameQuantType($0,$1)">>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 4.8K bytes - Viewed (0)