Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 174 for onStop (0.18 sec)

  1. tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h

        const auto attr = DenseIntElementsAttr::get(shape_type, values);
        return builder.create<TF::ConstOp>(loc, attr);
      }
    
      const auto type = RankedTensorType::get(shape, builder.getF32Type());
      const auto value_attr = DenseFPElementsAttr::get(type, values);
      return builder.create<TF::ConstOp>(loc, value_attr);
    }
    
    // Creates a 1D array with integer/float type.
    template <typename T>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc

      Type resultType = RankedTensorType::get(shape, element_type);
      auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape);
      auto shape_tensor =
          rewriter.create<TF::ConstOp>(loc, shape_spec_type, constant_attr);
      return rewriter.create<TF::ReshapeOp>(loc, resultType, /*tensor=*/value,
                                            /*shape=*/shape_tensor);
    }
    
    template <typename BatchMatMulOpType>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

    }
    
    TF::ConstOp Create1DConstantOp(const std::vector<int>& value, Location loc,
                                   OpBuilder* builder) {
      auto type =
          mlir::RankedTensorType::get(value.size(), builder->getIntegerType(32));
      auto dense_values = mlir::DenseIntElementsAttr::get(type, value);
      return builder->create<TF::ConstOp>(loc, dense_values);
    }
    
    TF::ConstOp CreateScalarConstantOp(int value, Location loc,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
  4. pkg/scheduler/framework/plugins/tainttoleration/taint_toleration_test.go

    	tests := []struct {
    		name       string
    		pod        *v1.Pod
    		node       *v1.Node
    		wantStatus *framework.Status
    	}{
    		{
    			name: "A pod having no tolerations can't be scheduled onto a node with nonempty taints",
    			pod:  podWithTolerations("pod1", []v1.Toleration{}),
    			node: nodeWithTaints("nodeA", []v1.Taint{{Key: "dedicated", Value: "user1", Effect: "NoSchedule"}}),
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Jun 12 13:26:09 UTC 2024
    - 17.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/tests/unfreeze_constants.mlir

        return %cst_0 : tensor<8xf32>
      }
    }
    
    // -----
    
    // Tests the case when there is no ConstOp.
    module attributes {tf_saved_model.semantics} {
    
    // Check that nothing happens when there's no ConstOp in the graph.
    // CHECK-NOT: "tf_saved_model.session_initializer"()
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 17.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold_test.cc

      ASSERT_THAT(test_func, NotNull());
    
      Operation* mul_op = FindOperationOfType<TF::MulOp>(test_func);
      SmallVector<Value> results = ConstantFoldOpIfPossible(mul_op);
      EXPECT_THAT(results, SizeIs(1));
      EXPECT_TRUE(isa<TF::ConstOp>(results[0].getDefiningOp()));
    }
    
    TEST_F(ConstantFoldingTest, NotFoldingIdentity) {
      constexpr absl::string_view kModuleCode = R"mlir(
        module {
          func.func @test_fold_constant() -> (tensor<1024x24x24x3xf32>) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 04 07:19:09 UTC 2024
    - 10.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir

      // CHECK-NEXT: [[PREFIX:%.*]] = tf_mlrt.constop
      %cst = "tf.Const"() {__op_key = 0: i32, value = dense<"restore_ariables"> : tensor<!tf_type.string>} : () -> tensor<!tf_type.string>
      // CHECK-NEXT: [[SLICE:%.*]] = tf_mlrt.constop
      %cst_0 = "tf.Const"()  {__op_key = 1: i32, value = dense<""> : tensor<1x!tf_type.string>} : () -> tensor<1x!tf_type.string>
      // CHECK-NEXT: [[NAME:%.*]] = tf_mlrt.constop
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 20:44:15 UTC 2024
    - 24.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc

        IntegerAttr group_key = rewriter.getI32IntegerAttr(0);
    
        auto const_group_size = rewriter.create<TF::ConstOp>(
            assign_group->getLoc(), assign_group.getResult(0).getType(),
            group_size);
        auto const_group_key = rewriter.create<TF::ConstOp>(
            assign_group->getLoc(), assign_group.getResult(1).getType(), group_key);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16K bytes
    - Viewed (0)
  9. staging/src/k8s.io/apiserver/pkg/util/proxy/streamtranslator_test.go

    	spdyServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
    		ctx, err := createSPDYServerStreams(w, req, Options{
    			Stdin:  true,
    			Stdout: true,
    		})
    		if err != nil {
    			t.Errorf("error on createHTTPStreams: %v", err)
    			return
    		}
    		defer ctx.conn.Close()
    		// Loopback STDIN data onto STDOUT stream.
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue Feb 27 23:21:55 UTC 2024
    - 36.4K bytes
    - Viewed (0)
  10. pkg/scheduler/framework/plugins/nodeaffinity/node_affinity_test.go

    			runPreFilter: true,
    		},
    		{
    			name: "Pod with no Affinity will schedule onto a node",
    			pod:  &v1.Pod{},
    			labels: map[string]string{
    				"foo": "bar",
    			},
    			wantPreFilterStatus: framework.NewStatus(framework.Skip),
    			runPreFilter:        true,
    		},
    		{
    			name: "Pod with Affinity but nil NodeSelector will schedule onto a node",
    			pod: &v1.Pod{
    				Spec: v1.PodSpec{
    					Affinity: &v1.Affinity{
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon Dec 18 12:00:10 UTC 2023
    - 38.7K bytes
    - Viewed (0)
Back to top