Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetTestGraph (0.1 sec)

  1. tensorflow/compiler/jit/tests/device_compiler_test_helper.h

      }
    
      JitCompilationListener* listener() const { return listener_; }
    
      // Returns a test graph that will split into two XLA clusters (due to a node
      // with _XlaCompile = false).
      GraphDef GetTestGraph(const PartialTensorShape& input_shape);
    
      // Runs the graph using specified batch size both with and without XLA JIT
      // compilation. Returns an error if the results between the two do not match.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 08:24:16 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/tests/device_compiler_test_helper.cc

      for (const auto& attr : attrs)
        node.mutable_attr()->insert({attr.first, attr.second.proto});
      return node;
    }
    
    }  // namespace
    
    GraphDef DeviceCompilerSerializeTest::GetTestGraph(
        const PartialTensorShape& input_shape) {
      FunctionDef make_test_fn = FunctionDefHelper::Define(
          "TestFn", {"a:float", "b:float", "c:float"}, {"m:float"}, {},
          {{{"d"}, "Add", {"a", "b"}, {{"T", DT_FLOAT}}},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 08:24:16 UTC 2024
    - 6.2K bytes
    - Viewed (0)
Back to top