Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 60 of 117 for get_attr (0.14 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

        func::FuncOp ResolveFunc(::mlir::SymbolTableCollection* table) {
          if (table)
            return table->lookupNearestSymbolFrom<func::FuncOp>(*this, getFAttr());
          return SymbolTable::lookupNearestSymbolFrom<func::FuncOp>(*this, getFAttr());
        }
        // TODO(b/204997177): Deprecate and remove.
        func::FuncOp func() {  return ResolveFunc(nullptr); }
      }];
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc

        auto range = llvm::make_filter_range(op->getUsers(), [&](Operation *user) {
          bool same_block = user->getBlock() == op->getBlock();
          bool same_device = op->getAttr(kDeviceAttr) == user->getAttr(kDeviceAttr);
          return same_block && same_device && state.IsMember(user);
        });
        return {range.begin(), range.end()};
      }
    
      // Users of block argument must be in the same block.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 27.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/reduce_type_precision.cc

            op.getLoc(),
            /*result=*/
            mlir::cast<TensorType>(op.getResult().getType())
                .clone(builder.getI4Type()),
            /*operand=*/op.getOperands(), op->getAttrs());
        rewriter.replaceAllUsesWith(op.getResult(), new_gather_op.getResult());
    
        return success();
      }
    };
    
    class ReduceTypePrecisionPass
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc

        // OperationState so we can add regions to the new up.
        OperationState state(op->getLoc(), op->getName().getStringRef(), operands,
                             new_results, op->getAttrs(), op->getSuccessors());
        for (Region &region : op->getRegions()) {
          Region &new_region = *state.addRegion();
          rewriter.inlineRegionBefore(region, new_region, new_region.begin());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_read_for_write.cc

        auto loc = cluster_func.getLoc();
        auto new_cluster_func = builder.create<tf_device::ClusterFuncOp>(
            loc, cluster_func.getResultTypes(), operands, cluster_func->getAttrs());
        cluster_func.replaceAllUsesWith(new_cluster_func);
        func::FuncOp func = cluster_func.getFuncOp();
        Block& block = func.front();
        for (Value read_operand : read_operands)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 16:54:40 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  6. staging/src/k8s.io/apiserver/pkg/storage/testing/store_tests.go

    			// doesn't automatically preclude some scenarios from happening.
    			t.Parallel()
    
    			if ignoreWatchCacheTests && tt.ignoreForWatchCache {
    				t.Skip()
    			}
    
    			if tt.pred.GetAttrs == nil {
    				tt.pred.GetAttrs = getAttrs
    			}
    
    			out := &example.PodList{}
    			storageOpts := storage.ListOptions{
    				ResourceVersion:      tt.rv,
    				ResourceVersionMatch: tt.rvMatch,
    				Predicate:            tt.pred,
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue Jun 11 12:45:33 UTC 2024
    - 91.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc

        }
        OpBuilder builder(call);
        auto new_call = builder.create<CallOp>(
            call.getLoc(), info.decomposed_callee.getFunctionType().getResults(),
            new_operands, call->getAttrs());
        new_call->setAttr(
            "f", SymbolRefAttr::get(
                     builder.getContext(),
                     const_cast<func::FuncOp&>(info.decomposed_callee).getName()));
        for (int64_t i = 0; i < call.getNumResults(); ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc

        op->setAttr("f", SymbolRefAttr::get(rewriter.getContext(),
                                            new_float_func.getName()));
    
        return success();
      }
    
      LogicalResult matchAndRewrite(TF::PartitionedCallOp op,
                                    PatternRewriter& rewriter) const override {
        const auto f_attr = mlir::dyn_cast<FlatSymbolRefAttr>(op.getFAttr());
        // Non-quantizable op
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc

          return true;
        }
      }
    
      SymbolTable symbol_table(op->getParentOfType<ModuleOp>());
      // Check the functions associated to CaseOp, IfOp and WhileOp.
      for (const NamedAttribute& attr : op->getAttrs()) {
        FlatSymbolRefAttr symbol_attr =
            dyn_cast_or_null<FlatSymbolRefAttr>(attr.getValue());
        if (!symbol_attr) continue;
    
        func::FuncOp target_func = dyn_cast_or_null<func::FuncOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  10. staging/src/k8s.io/apiextensions-apiserver/pkg/registry/customresource/strategy_test.go

    		t.Run(tc.name, func(t *testing.T) {
    			strategy := customResourceStrategy{selectableFieldSet: prepareSelectableFields(tc.selectableFields)}
    
    			_, fields, err := strategy.GetAttrs(tc.obj)
    			if err != nil {
    				t.Fatal(err)
    			}
    			if !reflect.DeepEqual(tc.expectFields, fields) {
    				t.Errorf("Expected fields '%+#v' but got '%+#v'", tc.expectFields, fields)
    			}
    		})
    	}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Apr 24 18:25:29 UTC 2024
    - 8.2K bytes
    - Viewed (0)
Back to top