Searched refs:computation_name (Results 1 – 8 of 8) sorted by relevance
32 explicit HumanReadableProfileBuilder(absl::string_view computation_name, in HumanReadableProfileBuilder() argument36 : computation_name_(computation_name), in HumanReadableProfileBuilder()
531 // TODO(b/239098765): Remove instruction_name and computation_name.532 string computation_name = 1 [deprecated = true]; field594 string computation_name = 3; field
307 const std::string& computation_name);1068 const std::string& computation_name) { in ParseInstructionList() argument1070 HloComputation::Builder builder(computation_name); in ParseInstructionList()
81 XlaComputation CreateTupleCeilComputation(const std::string& computation_name, in CreateTupleCeilComputation() argument83 XlaBuilder builder(computation_name); in CreateTupleCeilComputation()104 const std::string& computation_name, const Shape& tuple_shape) { in CreateTupleFloorComputation() argument105 XlaBuilder builder(computation_name); in CreateTupleFloorComputation()125 XlaComputation CreateTupleAddComputation(const std::string& computation_name, in CreateTupleAddComputation() argument127 XlaBuilder builder(computation_name); in CreateTupleAddComputation()145 XlaComputation CreateTupleSubComputation(const std::string& computation_name, in CreateTupleSubComputation() argument147 XlaBuilder builder(computation_name); in CreateTupleSubComputation()
46 computation_name = ops.get_default_graph().unique_name('computation')47 pivot = control_flow_ops.no_op(name=computation_name + '/pivot')48 return xla.XLACompileContext(name=computation_name, pivot=pivot)
260 std::string computation_name = in ImportAsFunc() local270 TF_RET_CHECK(!module_.lookupSymbol<FuncOp>(computation_name)) in ImportAsFunc()282 computation_name, func_type, attrs); in ImportAsFunc()283 auto visibility = computation_name == "main" ? FuncOp::Visibility::Public in ImportAsFunc()
280 XlaBuilder::XlaBuilder(const std::string& computation_name) in XlaBuilder() argument281 : name_(computation_name) {} in XlaBuilder()3828 const std::string& computation_name) { in CreateSubBuilder() argument3829 auto sub_builder = std::make_unique<XlaBuilder>(computation_name); in CreateSubBuilder()
178 explicit XlaBuilder(const std::string& computation_name);283 const std::string& computation_name);