Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for ret_attrs (0.15 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      // construction.
      llvm::SmallVector<mlir::NamedAttrList, 4> arg_attrs;
      arg_attrs.resize(func.getNumArguments());
      llvm::SmallVector<mlir::NamedAttrList, 4> ret_attrs;
      ret_attrs.resize(func.getNumResults());
    
      auto set_attributes_on_func = [&](Node* node, int64_t index, bool is_arg) {
        for (const auto& node_attr : node->attrs()) {
          const auto& key = node_attr.first;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

          auto input_iter = input_attrs.begin() + outer * inner_size;
          for (int64_t inner = 0; inner < inner_size; ++inner)
            out_attrs[out++] = *input_iter++;
        }
      }
    
      return DenseElementsAttr::get(output_type, out_attrs);
    }
    
    }  // end anonymous namespace
    
    LogicalResult ConcatenationOp::verify() {
      ConcatenationOp op = *this;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      llvm::SmallVector<mlir::DictionaryAttr, 4> arg_attrs, res_attrs;
      main_op.getAllArgAttrs(arg_attrs);
      main_op.getAllResultAttrs(res_attrs);
      std::vector<std::string> sig_def_inputs =
          GetStringsFromDictionaryAttr(arg_attrs, kTfSavedModelIndexPathAttr);
      std::vector<std::string> sig_def_outputs =
          GetStringsFromDictionaryAttr(res_attrs, kTfSavedModelIndexPathAttr);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  4. staging/src/k8s.io/apiserver/pkg/registry/generic/registry/store_test.go

    	}
    	return storage.SelectionPredicate{
    		Label:    labels.Everything().Add(*l),
    		Field:    fields.Everything(),
    		GetAttrs: getPodAttrs,
    	}
    }
    
    func matchEverything() storage.SelectionPredicate {
    	return storage.SelectionPredicate{
    		Label: labels.Everything(),
    		Field: fields.Everything(),
    		GetAttrs: func(obj runtime.Object) (label labels.Set, field fields.Set, err error) {
    			return nil, nil, nil
    		},
    	}
    }
    
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Jun 12 10:12:02 UTC 2024
    - 101.8K bytes
    - Viewed (0)
  5. pkg/registry/batch/job/strategy_test.go

    	}
    	for name, tc := range cases {
    		t.Run(name, func(t *testing.T) {
    			if tc.job == nil {
    				_, _, err := GetAttrs(tc.nonJobObject)
    				if diff := cmp.Diff(tc.wantErr, err.Error()); diff != "" {
    					t.Errorf("Unexpected errors (-want,+got):\n%s", diff)
    				}
    			} else {
    				gotLabels, _, err := GetAttrs(tc.job)
    				if err != nil {
    					t.Errorf("Error %s supposed to be nil", err.Error())
    				}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Apr 24 18:25:29 UTC 2024
    - 101.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

        auto tf_fused_batch_norm_op_v3 = CreateTfOp<TF::FusedBatchNormV3Op>(
            rewriter, tf_fused_batch_norm_op, new_result_types,
            tf_fused_batch_norm_op.getOperands(),
            tf_fused_batch_norm_op->getAttrs());
    
        rewriter.replaceOp(tf_fused_batch_norm_op,
                           tf_fused_batch_norm_op_v3->getResults().drop_back());
        return success();
      }
    };
    }  // namespace.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          auto name = attr.getName();
          return name == "call_target_name" || name == "backend_config" ||
                 name == "api_version" || name == "called_computations";
        };
        for (const auto& attr : op->getAttrs()) {
          if (!is_supported_attr_name(attr)) {
            return op.emitOpError()
                   << attr.getName().getValue()
                   << " is not a supported attribute for ApproxTopK";
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

        // Create new case/if region op.
        auto new_op = CreateTfOp<CaseOrIfRegionOp>(rewriter, op, new_result_types,
                                                   op.getOperand(), op->getAttrs(),
                                                   op.getNumRegions());
    
        int next_index = 0;
        for (auto result : op.getResults()) {
          if (!result_to_extern_value.count(result)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
Back to top