Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 70 of 96 for makeBody (0.24 sec)

  1. tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc

        }
      }
    
      auto new_while_op = OpBuilder(while_op).create<WhileOp>(
          while_op.getLoc(), new_result_types, new_operands, while_op->getAttrs());
      new_while_op.getCond().takeBody(while_op.getCond());
      new_while_op.getBody().takeBody(while_op.getBody());
    
      for (auto i = 0; i < n; ++i) {
        if (!while_op.getResult(i).use_empty()) {
          auto j = id_map[i];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/raise_custom_ops.cc

          inner_op->setOperand(idx_args.index(), idx_args.value());
        }
        custom_op->setAttrs(inner_op->getAttrs());
        builder.create<YieldOp>(loc, inner_op->getResults());
        custom_op.getBody().takeBody(region);
    
        op->replaceAllUsesWith(custom_op);
        op->erase();
      }
    }
    }  // namespace
    
    std::unique_ptr<OperationPass<func::FuncOp>> CreateRaiseCustomOpsPass() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 4.3K bytes
    - Viewed (0)
  3. pkg/scheduler/framework/plugins/helper/spread_test.go

    				},
    			},
    		}
    		services = append(services, service)
    		fakeInformerFactory.Core().V1().Services().Informer().GetStore().Add(service)
    	}
    	var pods []*v1.Pod
    	for i := 0; i < 5; i++ {
    		pod := st.MakePod().Name(fmt.Sprintf("test-pod-%d", i)).
    			Namespace("test").
    			Label("app", fmt.Sprintf("test-%d", i)).
    			Label("label", fmt.Sprintf("label-%d", i)).
    			Obj()
    		pods = append(pods, pod)
    	}
    
    	tests := []struct {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Thu May 05 17:48:55 UTC 2022
    - 2.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

      auto new_while = builder.create<TF::WhileRegionOp>(
          while_op.getLoc(), body_region.front().getTerminator()->getOperandTypes(),
          new_while_operands, while_op->getAttrs());
      new_while.getBody().takeBody(body_region);
      new_while.getCond().takeBody(cond_region);
      for (const auto& entry : output_buffer_to_size) {
        (*buffer_to_size)[new_while.getResult(std::get<0>(entry))] = {
            new_while.getResult(std::get<1>(entry)), std::get<2>(entry)};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  5. pkg/scheduler/framework/plugins/dynamicresources/dynamicresources_test.go

    		return pod
    	}()
    	podWithTwoClaimNames = st.MakePod().Name(podName).Namespace(namespace).
    				UID(podUID).
    				PodResourceClaims(v1.PodResourceClaim{Name: resourceName, Source: v1.ClaimSource{ResourceClaimName: &claimName}}).
    				PodResourceClaims(v1.PodResourceClaim{Name: resourceName2, Source: v1.ClaimSource{ResourceClaimName: &claimName2}}).
    				Obj()
    
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon May 27 07:57:10 UTC 2024
    - 61.9K bytes
    - Viewed (0)
  6. pkg/controller/garbagecollector/dump_test.go

    				{F: types.UID("foxtrot"), T: types.UID("golf")},
    			},
    		},
    		{
    			file: "escaping.dot",
    			nodes: []*dotVertex{
    				NewDOTVertex(makeNode(ref1, withOwners(ref2))),
    				NewDOTVertex(makeNode(ref2)),
    			},
    			edges: []dotEdge{
    				{F: types.UID(ref1.UID), T: types.UID(ref2.UID)},
    			},
    		},
    	}
    
    	for _, tc := range testcases {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri Apr 14 00:05:53 UTC 2023
    - 11.2K bytes
    - Viewed (0)
  7. pkg/scheduler/framework/types_test.go

    				t.Errorf("expected: %#v, got: %#v", test.expected, test.resource)
    			}
    		})
    	}
    }
    
    func TestNewNodeInfo(t *testing.T) {
    	nodeName := "test-node"
    	pods := []*v1.Pod{
    		st.MakePod().UID("test-1").Namespace("node_info_cache_test").Name("test-1").Node(nodeName).
    			Containers([]v1.Container{st.MakeContainer().ResourceRequests(map[v1.ResourceName]string{
    				v1.ResourceCPU:    "100m",
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Apr 24 18:25:29 UTC 2024
    - 45.9K bytes
    - Viewed (0)
  8. pkg/scheduler/testing/wrappers.go

    func (p *PodWrapper) Overhead(rl v1.ResourceList) *PodWrapper {
    	p.Spec.Overhead = rl
    	return p
    }
    
    // NodeWrapper wraps a Node inside.
    type NodeWrapper struct{ v1.Node }
    
    // MakeNode creates a Node wrapper.
    func MakeNode() *NodeWrapper {
    	w := &NodeWrapper{v1.Node{}}
    	return w.Capacity(nil)
    }
    
    // Obj returns the inner Node.
    func (n *NodeWrapper) Obj() *v1.Node {
    	return &n.Node
    }
    
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon May 27 07:57:10 UTC 2024
    - 42.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc

      tf_device::LaunchOp launch_op = builder->create<tf_device::LaunchOp>(
          builder->getUnknownLoc(), builder->getStringAttr(c.target),
          live_out_types);
    
      // Attach the region to launch_op.
      launch_op.getBody().takeBody(region);
    
      // Replace any external uses of live-out values with return values of launch
      // op. So live-out values no longer escape the region.
      ReplaceLiveOutExternalUses(live_outs, launch_op);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 05 13:30:21 UTC 2023
    - 6.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/hoist_replicate_invariant_resource_writes.cc

          replicate_op->getLoc(), new_result_types, replicate_op->getOperands(),
          replicate_op->getAttrs());
    
      // Move region to the new op.
      new_replicate_op.getRegion().takeBody(replicate_op.getRegion());
    
      // Replace all old uses with new op results.
      int old_num_results = replicate_op->getNumResults();
      replicate_op->replaceAllUsesWith(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 5.8K bytes
    - Viewed (0)
Back to top