| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_util.cc | 206 HloComputation* computation, const WhileUtil::LoopStateTy& init_values) { in MakeInitTupleFromInitValues() argument
208 init_values_with_indvar.reserve(init_values.size() + 1); in MakeInitTupleFromInitValues()
212 absl::c_copy(init_values, std::back_inserter(init_values_with_indvar)); in MakeInitTupleFromInitValues()
217 static Shape MakeLoopStateShape(const WhileUtil::LoopStateTy& init_values) { in MakeLoopStateShape() argument
219 loop_state_shape_components.reserve(init_values.size() + 1); in MakeLoopStateShape()
221 absl::c_transform(init_values, in MakeLoopStateShape()
229 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop() argument
234 Shape loop_state_shape = MakeLoopStateShape(init_values); in MakeCountedLoop()
242 MakeInitTupleFromInitValues(computation, init_values)); in MakeCountedLoop()
251 for (int64 i = 0, e = init_values.size(); i < e; i++) { in MakeCountedLoop()
|
| D | while_util.h | 81 const LoopStateTy& init_values,
|
| D | hlo_instruction.cc | 246 auto init_values = in CreateFromProto() local
250 CreateReduce(shape, inputs, init_values, in CreateFromProto()
1043 absl::Span<HloInstruction* const> init_values, in CreateReduce() argument
1049 all_args.insert(all_args.end(), init_values.begin(), init_values.end()); in CreateReduce()
|
| D | hlo_instruction.h | 625 absl::Span<HloInstruction* const> init_values,
|
| D | shape_inference.cc | 2080 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape() local
2085 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply, init_values, element_types, in InferReduceShape()
|
| D | hlo_instructions.h | 474 absl::Span<HloInstruction* const> init_values() const { in init_values() function
|
| D | algebraic_simplifier.cc | 2916 auto init_value = reduce->init_values()[0]; in HandleReduce()
2930 reduce->init_values()[i], {}))); in HandleReduce()
|
| D | hlo_evaluator_typed_visitor.h | 1711 &parent_->GetEvaluatedLiteralFor(reduce->init_values()[i]); in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | vector_support_library.cc | 339 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeHorizontalSums() argument
344 return ComputeAvxOptimizedHorizontalSums(std::move(vectors), init_values); in ComputeHorizontalSums()
350 if (init_values) { in ComputeHorizontalSums()
353 b()->CreateExtractElement(init_values, b()->getInt32(i))); in ComputeHorizontalSums()
361 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeAvxOptimizedHorizontalSums() argument
376 if (init_values) { in ComputeAvxOptimizedHorizontalSums()
377 low = AddInternal(ExtractLowHalf(init_values), low); in ComputeAvxOptimizedHorizontalSums()
381 if (init_values) { in ComputeAvxOptimizedHorizontalSums()
382 high = AddInternal(ExtractHighHalf(init_values), high); in ComputeAvxOptimizedHorizontalSums()
|
| D | vector_support_library.h | 219 std::vector<llvm::Value*> vectors, llvm::Value* init_values = nullptr);
267 std::vector<llvm::Value*> vectors, llvm::Value* init_values);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 513 std::vector<xla::XlaOp> init_values; in Compile() local
514 init_values.reserve(4); in Compile()
515 init_values.push_back(xla::ConstantR0<int32>(builder, 0)); // col_idx in Compile()
516 init_values.push_back(xla::ConstantR0<int32>(builder, 0)); // num_outputs in Compile()
517 init_values.push_back(iou_thresh_mask); in Compile()
518 init_values.push_back(included_iou); in Compile()
522 SuppressBodyFn(num_boxes), init_values, in Compile()
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow.py | 277 init_values = get_state()
279 set_state(init_values)
|
| /external/antlr/runtime/Ruby/lib/antlr3/ |
| D | profile.rb | 109 init_values = Array.new( self.class.members.length, 0 )
110 super( *init_values )
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 1311 std::vector<T> init_values) { in MkTensor() argument
1316 x.flat<T>()(i) = init_values[vi++]; in MkTensor()
1317 if (vi >= init_values.size()) vi = 0; in MkTensor()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 434 init_values = [True, pfor_input.pfor.all_indices] + inputs + output_tas
441 return init_values, inputs_stacked, shape_invariants
584 init_values, inputs_stacked, shape_invariants = self._create_init_values(
644 cond, body, init_values, shape_invariants=shape_invariants)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 445 absl::Span<const XlaOp> init_values,
866 absl::Span<const XlaOp> init_values,
1538 absl::Span<const XlaOp> init_values,
|
| D | xla_builder.cc | 1947 absl::Span<const XlaOp> init_values, in Reduce() argument
1958 all_operands.insert(all_operands.end(), init_values.begin(), in Reduce()
1959 init_values.end()); in Reduce()
3198 absl::Span<const XlaOp> init_values, in Reduce() argument
3201 return builder->Reduce(operands, init_values, computation, in Reduce()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1664 <b> `Reduce(operands..., init_values..., computation, dimensions)` </b>
1669 | `init_values` | Sequence of N `XlaOp` | N scalars of types `T_0, ..., T_N`. |
|