Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 166 for setAttr (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc

            /*config=*/builder.getStringAttr(""),
            /*config_proto=*/builder.getStringAttr(""),
            /*executor_type=*/builder.getStringAttr(""));
        call_op->setAttr("device", launch->getAttrOfType<StringAttr>("device"));
        launch.replaceAllUsesWith(call_op);
        launch.erase();
      });
    }
    
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 31 21:08:09 UTC 2023
    - 2.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc

          "bad_consumers",
          b.getI32ArrayAttr(llvm::ArrayRef<int32_t>(
              versions.bad_consumers().data(),
              versions.bad_consumers().data() + versions.bad_consumers().size())));
      module->setAttr("tf.versions",
                      b.getDictionaryAttr(llvm::ArrayRef<mlir::NamedAttribute>(
                          {producer, min_consumer, bad_consumers})));
    }
    
    mlir::LogicalResult ExtractTfVersions(mlir::ModuleOp module,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/debug/debug_test.cc

     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MutatePass)
    
      void runOnOperation() override {
        mlir::OpBuilder builder(&getContext());
        getOperation()->setAttr("tfl.random_attr", builder.getUnitAttr());
      }
    };
    
    class AlwaysFailPass
        : public mlir::PassWrapper<AlwaysFailPass, mlir::OperationPass<>> {
     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(AlwaysFailPass)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 11:15:16 UTC 2024
    - 9.7K bytes
    - Viewed (0)
  4. tensorflow/c/python_api.cc

      mutex_lock l(graph->mu);
      RecordMutation(graph, op, mutation_type);
    }
    
    void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input) {
      TF_AddOperationControlInput(graph, op, input);
    }
    
    void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
                 TF_Buffer* attr_value_proto, TF_Status* status) {
      TF_SetAttr(graph, op, attr_name, attr_value_proto, status);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jul 12 18:48:56 UTC 2023
    - 3.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc

      }
    
      auto init = builder.create<func::FuncOp>(
          module.getLoc(), name,
          FunctionType::get(module.getContext(), /*inputs=*/{}, /*results=*/{}));
    
      // Make savedmodel verification happy.
      init->setAttr("tf_saved_model.exported_names",
                    builder.getStrArrayAttr({name}));
    
      builder.setInsertionPointToStart(init.addEntryBlock());
    
      for (func::FuncOp func : tf_saved_model::GetInitializerFunctions(module)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 01 05:03:09 UTC 2023
    - 3.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/stablehlo/passes/restore_function_name.cc

        return;
      }
    
      auto function =
          symbol_table.lookup<func::FuncOp>(current_function_name.getValue());
      if (function) {
        function.setName(original_function_name);
      }
    
      call_op->setAttr(TF::kStablehloEntryFunctionAttrName,
                       FlatSymbolRefAttr::get(original_function_name));
    }
    
    void RestoreFunctionNamePass::runOnOperation() {
      ModuleOp module_op = getOperation();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 08:32:43 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/constant_op_device_assignment.cc

          // device of the user operation.
          if (cloned_op_by_device.find(device_attr.getValue()) ==
              cloned_op_by_device.end()) {
            mlir::Operation *new_op = builder.clone(*op.getOperation());
            new_op->setAttr(kDeviceAttr, device_attr);
            cloned_op_by_device[device_attr.getValue()] = new_op;
          }
          // Update the user operation to use the result of the cloned ConstOp.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 3.3K bytes
    - Viewed (0)
  8. src/test/java/org/codelibs/fess/util/PrunedTagTest.java

        }
    
        public void test_toString() {
            String tag = "tag", id = "id", css = "css", attrName = "attrName", attrValue = "attrValue";
            PrunedTag prunedtag = new PrunedTag(tag);
            prunedtag.setAttr(attrName, attrValue);
            prunedtag.setId(id);
            prunedtag.setCss(css);
            assertEquals("PrunedTag [tag=" + tag + ", id=" + id + ", css=" + css + ", attrName=" + attrName + ", attrValue=" + attrValue + "]",
    Registered: Wed Jun 12 13:08:18 UTC 2024
    - Last Modified: Thu Feb 22 01:37:57 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/mark_initialized_variables.cc

        } else {
          auto handle = resource_tensor.scalar<tensorflow::ResourceHandle>()();
          is_variable_initialized =
              IsVariableInitialized(var_op, handle.device(), mgr, session);
        }
        var_op->setAttr("_is_initialized",
                        BoolAttr::get(context, is_variable_initialized));
      }
      return success();
    }
    
    LogicalResult MarkInitializedVariablesInFunction(ModuleOp module,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 13 19:14:56 UTC 2023
    - 4.1K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/experimental/tac/tflite_import_export.cc

          auto float_attr =
              mlir::FloatAttr::get(float_type, static_cast<float>(cost));
          device_costs.push_back({device_identifier, float_attr});
        }
    
        op->setAttr("per_device_costs",
                    mlir::DictionaryAttr::get(module.getContext(), device_costs));
      });
    }
    
    }  // namespace
    
    //////////// Importer ////////////
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 4.6K bytes
    - Viewed (0)
Back to top