Searched refs:computation_name (Results 1 – 9 of 9) sorted by relevance
32 explicit HumanReadableProfileBuilder(absl::string_view computation_name, in HumanReadableProfileBuilder() argument36 : computation_name_(computation_name), in HumanReadableProfileBuilder()
354 string computation_name = 1; field415 string computation_name = 3; field
119 const string& computation_name);594 const string& computation_name) { in ParseInstructionList() argument596 HloComputation::Builder builder(computation_name); in ParseInstructionList()
79 XlaComputation CreateTupleCeilComputation(const string& computation_name, in CreateTupleCeilComputation() argument81 XlaBuilder builder(computation_name); in CreateTupleCeilComputation()101 XlaComputation CreateTupleFloorComputation(const string& computation_name, in CreateTupleFloorComputation() argument103 XlaBuilder builder(computation_name); in CreateTupleFloorComputation()123 XlaComputation CreateTupleAddComputation(const string& computation_name, in CreateTupleAddComputation() argument125 XlaBuilder builder(computation_name); in CreateTupleAddComputation()143 XlaComputation CreateTupleSubComputation(const string& computation_name, in CreateTupleSubComputation() argument145 XlaBuilder builder(computation_name); in CreateTupleSubComputation()
54 computation_name = ops.get_default_graph().unique_name('computation')55 pivot = control_flow_ops.no_op(name=computation_name + '/pivot')56 return xla.XLACompileContext(name=computation_name, pivot=pivot)
413 ComputationBuilder::ComputationBuilder(const string& computation_name) in ComputationBuilder() argument414 : builder_(computation_name) {} in ComputationBuilder()
208 ComputationBuilder(const string& computation_name);
121 XlaBuilder::XlaBuilder(const string& computation_name) in XlaBuilder() argument122 : name_(computation_name) {} in XlaBuilder()2638 const string& computation_name) { in CreateSubBuilder() argument2639 auto sub_builder = absl::make_unique<XlaBuilder>(computation_name); in CreateSubBuilder()
137 XlaBuilder(const string& computation_name);198 std::unique_ptr<XlaBuilder> CreateSubBuilder(const string& computation_name);