Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for mutable_attr (0.23 sec)

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

                (*node_def.mutable_attr())["shared_name"].set_s(node_def.name());
              } else {
                // Use the concat of function name and node name for such ops in a
                // function as the shared_name. "@" is used as the separator because
                // it is not allowed in the function name or the node name.
                (*node_def.mutable_attr())["shared_name"].set_s(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 27 14:33:47 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc

                                      }}},
                                    {{"c", "t0:y:0"}});
    
      AttrValue true_attribute;
      true_attribute.set_b(true);
      (*called_func.mutable_attr())[kXlaMustCompileAttr] = true_attribute;
      *library.add_function() = called_func;
      TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library));
      FunctionLibraryDefinition flib_def(OpRegistry::Global(), library);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 4.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/tests/device_compiler_test_helper.cc

            attrs) {
      NodeDef node;
      node.set_name(std::string(name));
      node.set_op(std::string(op));
      for (const auto& input : inputs) node.add_input(input);
      for (const auto& attr : attrs)
        node.mutable_attr()->insert({attr.first, attr.second.proto});
      return node;
    }
    
    }  // namespace
    
    GraphDef DeviceCompilerSerializeTest::GetTestGraph(
        const PartialTensorShape& input_shape) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 08:24:16 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/device_compilation_cluster_signature_test.cc

    }
    
    void BM_BuildSignature(::testing::benchmark::State& state) {
      const int n_args = state.range(0);
    
      NameAttrList fn;
      fn.set_name("afunction");
      for (int i = 0; i < n_args; i++) {
        (*fn.mutable_attr())[absl::StrCat("T", i)].set_type(DT_FLOAT);
      }
      std::vector<XlaCompiler::Argument> args(n_args);
      for (int i = 0; i < n_args; i++) {
        args[i].kind = (((i % 3) == 0) ? XlaCompiler::Argument::kConstant
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 4.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc

      LOG_FIRST_N(INFO, 1) << "Run Node Expansion Passes";
    
      // Get the FunctionDef and insert that into the context
      const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef();
      auto& ctx = orig_op->EagerContext();
      Fprint128 cache_key =
          orig_op->MutableAttrs()->CacheKey(orig_op->DeviceName());
      // Include soft placement policy in cache key since the placement strategy
      // can change and thus affect which kernel is picked.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Feb 25 16:22:36 UTC 2024
    - 3.8K bytes
    - Viewed (0)
Back to top