Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for mutable_attr (0.11 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/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