Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 22 for op_names (0.31 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

          counted only if the ops are named 'op_name' and has 'attr_val' if
          'attr_name' is specified.
        """
        op_count = 0
        for op_name in op_names:
          # Check the main graph
          op_count += self._count_op_with_name_and_attribute(
              nodes=graphdef.node,
              op_name=op_name,
              attr_name=attr_name,
              attr_val=attr_val,
              get_op_name=get_op_name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir

      // CHECK: %[[RESULT1:.*]] = "tf.Abs"(%[[RESULT0]])
      // CHECK: %[[RESULT2:.*]] = "tf.Log"(%[[RESULT1]])
      // CHECK: return %[[RESULT2]]
    
      %0 = "tf._UnaryOpsComposition"(%arg0) {op_names = ["Asin", "Abs", "Log"]} : (tensor<4xf32>) -> tensor<4xf32>
      func.return %0 : tensor<4xf32>
    }
    
    
    // CHECK-LABEL: @round_int
    func.func @round_int(%arg0: tensor<2xi32>) -> tensor<2xi32> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jan 05 18:35:42 UTC 2024
    - 92K bytes
    - Viewed (0)
  3. tensorflow/c/kernels_test.cc

      const char* node_name = "SomeNodeName";
      const char* op_name = "FooOp";
      const char* device_name = "FakeDeviceName1";
    
      REGISTER_OP(op_name)
          .Input("input1: double")
          .Input("input2: uint8")
          .Output("output1: uint8")
          .Attr("SomeDataTypeAttr: type");
    
      TF_KernelBuilder* builder = TF_NewKernelBuilder(
          op_name, device_name, &MyCreateFunc, &MyComputeFunc, &MyDeleteFunc);
    
      {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

      def _visit_tfr_builtins(self, op_name, args, node):
        arg_strs = []
        arg_tys = []
        for arg in args:
          value, ty = self.visit(arg)
          arg_strs.append(value)
          arg_tys.append(ty)
        tfr_op_name = 'tfr.' + op_name[5:]
        ret_tys = (
            TFR_BUILTINS[op_name](*arg_tys)
            if callable(TFR_BUILTINS[op_name]) else TFR_BUILTINS[op_name])
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

            op_state.operands[1] = shape_op;
          }
        }
      }
      if (op_name == "vhlo.reduce_v1" || op_name == "vhlo.reduce_window_v1" ||
          op_name == "vhlo.sort_v1" || op_name == "vhlo.scatter_v1") {
        op_state.addRegion();
      }
      if (op_name == "vhlo.while_v1") {
        op_state.addRegion();
        op_state.addRegion();
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_test.cc

      TF_OperationDescription* init(string type) {
        // Construct op_name to match the name used by REGISTER_OP in the
        // ATTR_TEST_REGISTER calls above.
        string op_name = "CApiAttributesTestOp";
        if (type.find("list(") == 0) {
          op_name += "List";
          type = type.replace(0, 5, "");
          type = type.replace(type.size() - 1, 1, "");
        }
        op_name += type;
        return TF_NewOperation(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
  7. pkg/controller/disruption/disruption_test.go

    		failOnce.Do(func() {
    			// (B) Evict two pods and fail this update.
    			evict(podNames[1], podNames[2])
    			handled = true
    			err = errors.NewConflict(a.GetResource().GroupResource(), pdb.Name, fmt.Errorf("conflict"))
    		})
    		return handled, obj, err
    	})
    
    	// (A) Delete one pod
    	if err := dc.coreClient.CoreV1().Pods("default").Delete(ctx, podNames[0], metav1.DeleteOptions{}); err != nil {
    		t.Fatal(err)
    	}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Sat May 04 18:33:12 UTC 2024
    - 50K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

        for (StringRef op_name : func_count_map.keys()) {
          const int32_t quantized_count = func_count_map[op_name].num_quant;
          const int32_t total_count =
              quantized_count + func_count_map[op_name].num_float;
          lines.push_back(absl::StrFormat("%s %d/%d",
                                          pad_string(op_name, name_col_width),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  9. pkg/controller/volume/persistentvolume/pv_controller.go

    	pods, err := ctrl.findPodsByPVCKey(pvcKey)
    	if err != nil {
    		return nil, err
    	}
    	podNames := []string{}
    	for _, pod := range pods {
    		if util.IsPodTerminated(pod, pod.Status) {
    			continue
    		}
    		if len(pod.Spec.NodeName) == 0 {
    			podNames = append(podNames, pod.Name)
    		}
    	}
    	return podNames, nil
    }
    
    // doDeleteVolume finds appropriate delete plugin and deletes given volume, returning
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 10 08:42:31 UTC 2024
    - 89.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

    }
    
    bool IsInSubgraph(const string& func_id) { return !func_id.empty(); }
    
    Status Encapsulator::CopySubgraphNodes(
        absl::flat_hash_map<const Node*, Node*>* node_images) {
      for (Node* node : graph_in_->op_nodes()) {
        string func_id;
        TF_RETURN_IF_ERROR(GetFunctionNameAttr(node, &func_id));
        if (!IsInSubgraph(func_id)) continue;
    
        Subgraph& subgraph = subgraphs_[func_id];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
Back to top