Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for set_constant_value (0.18 sec)

  1. tensorflow/compiler/jit/node_matchers.h

        DCHECK(IsEmpty());
        op_ = std::move(op);
      }
    
      void set_assigned_device(string assigned_device) {
        DCHECK(IsEmpty());
        assigned_device_ = std::move(assigned_device);
      }
    
      void set_constant_value(Tensor constant_value) {
        DCHECK(IsEmpty());
        constant_value_ = std::move(constant_value);
        op_ = "Const";
      }
    
      void set_inputs(InputSeqMatcher inputs) {
        DCHECK(IsEmpty());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 8.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/node_matchers.cc

      return props;
    }
    
    NodeMatcherProperties ConstantValue(
        const ::tensorflow::Input::Initializer& val) {
      TF_CHECK_OK(val.status);
      NodeMatcherProperties props;
      props.set_constant_value(val.tensor);
      return props;
    }
    
    ::testing::Matcher<impl::OutEdge> Const(
        const ::tensorflow::Input::Initializer& val) {
      return Out(NodeWith(ConstantValue(val)));
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 03 16:15:20 UTC 2022
    - 16.8K bytes
    - Viewed (0)
Back to top