Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 60 of 119 for getRand (0.25 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

        // Input should be rank 4.
        if (!input_ty.hasRank() || input_ty.getRank() != 4) {
          return failure();
        }
    
        // Check that out_size is rank-1, length-2. Otherwise the size is not legal.
        if (!out_size_ty.hasRank() || out_size_ty.getRank() != 1 ||
            out_size_ty.getShape()[0] != 2) {
          return failure();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  10. 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)
Back to top