Searched defs:init_values (Results 1 – 8 of 8) sorted by relevance
206 HloComputation* computation, const WhileUtil::LoopStateTy& init_values) { in MakeInitTupleFromInitValues()217 static Shape MakeLoopStateShape(const WhileUtil::LoopStateTy& init_values) { in MakeLoopStateShape()229 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop()
474 absl::Span<HloInstruction* const> init_values() const { in init_values() function
246 auto init_values = in CreateFromProto() local1043 absl::Span<HloInstruction* const> init_values, in CreateReduce()
2080 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape() local
339 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeHorizontalSums()361 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeAvxOptimizedHorizontalSums()
513 std::vector<xla::XlaOp> init_values; in Compile() local
1311 std::vector<T> init_values) { in MkTensor()
1947 absl::Span<const XlaOp> init_values, in Reduce()3198 absl::Span<const XlaOp> init_values, in Reduce()