| /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()
221 const WhileUtil::LoopStateTy& init_values) { in MakeLoopStateShapeWithLayout() argument
223 loop_state_shape_components.reserve(init_values.size() + 1); in MakeLoopStateShapeWithLayout()
225 absl::c_transform(init_values, in MakeLoopStateShapeWithLayout()
239 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop() argument
247 Shape loop_state_shape = MakeLoopStateShapeWithLayout(init_values); in MakeCountedLoop()
255 MakeInitTupleFromInitValues(computation, init_values)); in MakeCountedLoop()
264 for (int64 i = 0, e = init_values.size(); i < e; i++) { in MakeCountedLoop()
|
| D | while_util.h | 81 const LoopStateTy& init_values,
|
| D | hlo_evaluator.cc | 2335 absl::Span<const Literal* const> init_values, in GenerateReduceOutputElement() argument
2344 bool use_fast_add = ShapeUtil::ElementIsFloating(init_values[0]->shape()) && in GenerateReduceOutputElement()
2356 results[i].CopyElementFrom(*init_values[i], {}, output_index)); in GenerateReduceOutputElement()
2360 double computed_result = *init_values[0]->GetAsDouble({}); in GenerateReduceOutputElement()
2405 absl::InlinedVector<const Literal*, 1> init_values(num_args); in HandleReduce() local
2409 init_values[i] = &GetEvaluatedLiteralFor(reduce->init_values()[i]); in HandleReduce()
2410 VLOG(3) << "HandleReduce init_literal: " << init_values[i]->ToString(); in HandleReduce()
2411 TF_RET_CHECK(ShapeUtil::IsScalar(init_values[i]->shape())); in HandleReduce()
2456 output_index, init_values, input_args, absl::Span<Literal>(results), in HandleReduce()
|
| D | hlo_instruction.cc | 247 auto init_values = in CreateFromProto() local
251 CreateReduce(shape, inputs, init_values, in CreateFromProto()
1138 absl::Span<HloInstruction* const> init_values, in CreateReduce() argument
1144 all_args.insert(all_args.end(), init_values.begin(), init_values.end()); in CreateReduce()
|
| D | hlo_instruction.h | 771 absl::Span<HloInstruction* const> init_values,
|
| D | shape_inference.cc | 2106 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape() local
2111 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply, init_values, element_types, in InferReduceShape()
|
| D | hlo_instructions.h | 521 absl::Span<HloInstruction* const> init_values() const { in init_values() function
|
| D | algebraic_simplifier.cc | 3558 auto init_value = reduce->init_values()[0]; in HandleReduce()
3572 reduce->init_values()[i], {}))); in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | vector_support_library.cc | 345 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeHorizontalSums() argument
350 return ComputeAvxOptimizedHorizontalSums(std::move(vectors), init_values); in ComputeHorizontalSums()
356 if (init_values) { in ComputeHorizontalSums()
359 b()->CreateExtractElement(init_values, b()->getInt32(i))); in ComputeHorizontalSums()
367 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeAvxOptimizedHorizontalSums() argument
382 if (init_values) { in ComputeAvxOptimizedHorizontalSums()
383 low = AddInternal(ExtractLowHalf(init_values), low); in ComputeAvxOptimizedHorizontalSums()
387 if (init_values) { in ComputeAvxOptimizedHorizontalSums()
388 high = AddInternal(ExtractHighHalf(init_values), high); in ComputeAvxOptimizedHorizontalSums()
|
| D | elemental_ir_emitter.cc | 146 for (const HloInstruction* instr : reduce_instr->init_values()) { in MakeElementGenerator()
|
| D | vector_support_library.h | 221 std::vector<llvm::Value*> vectors, llvm::Value* init_values = nullptr);
269 std::vector<llvm::Value*> vectors, llvm::Value* init_values);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 519 std::vector<xla::XlaOp> init_values; in Compile() local
520 init_values.reserve(4); in Compile()
521 init_values.push_back(xla::ConstantR0<int32>(builder, 0)); // col_idx in Compile()
522 init_values.push_back(xla::ConstantR0<int32>(builder, 0)); // num_outputs in Compile()
523 init_values.push_back(iou_thresh_mask); in Compile()
524 init_values.push_back(included_iou); in Compile()
528 SuppressBodyFn(num_boxes), init_values, in Compile()
|
| /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/compiler/xla/service/mlir_gpu/ |
| D | lhlo_dialect_emitter.cc | 253 auto init_values = in HandleReduce() local
259 auto reduce_op = builder.create<lhlo::ReduceOp>(loc, inputs, init_values, in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | lhlo_legalize_to_gpu.cc | 97 for (auto pair : llvm::zip(reduce_op.init_values(), reduce_op.out())) { in matchAndRewrite()
|
| D | legalize_tf.cc | 309 WhileBodyFnType body_fn, ArrayRef<Value> init_values, in CreateWhile32() argument
312 int value_count = init_values.size() + 1; in CreateWhile32()
320 init_values_with_loop_iv.append(init_values.begin(), init_values.end()); in CreateWhile32()
387 final_values->reserve(init_values.size()); in CreateWhile32()
388 for (int i = 0, e = init_values.size(); i < e; ++i) in CreateWhile32()
2218 Value init_values[] = {init_value, index_init_value}; in matchAndRewrite() local
2224 llvm::ArrayRef<Value>(init_values), reduction_dimensions); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | lhlo_ops.td | 148 Variadic<LHLO_BufferOrTuple>:$init_values,
|
| D | hlo_ops.cc | 783 ValueRange operands, ValueRange init_values, in build() argument
792 build(builder, state, result_ty, operands, init_values, dimensions); in build()
|
| D | hlo_ops.td | 556 Variadic<HLO_TensorOrTuple>:$init_values,
564 "ValueRange init_values, DenseIntElementsAttr dimensions"
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 1316 std::vector<T> init_values) { in MkTensor() argument
1321 x.flat<T>()(i) = init_values[vi++]; in MkTensor()
1322 if (vi >= init_values.size()) vi = 0; in MkTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 491 absl::Span<const XlaOp> init_values,
905 absl::Span<const XlaOp> init_values,
1625 absl::Span<const XlaOp> init_values,
|
| D | xla_builder.cc | 1914 absl::Span<const XlaOp> init_values, in Reduce() argument
1925 all_operands.insert(all_operands.end(), init_values.begin(), in Reduce()
1926 init_values.end()); in Reduce()
3267 absl::Span<const XlaOp> init_values, in Reduce() argument
3270 return builder->Reduce(operands, init_values, computation, in Reduce()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 454 init_values = [True, pfor_input.pfor.all_indices] + inputs + output_tas
461 return init_values, inputs_stacked, shape_invariants
601 init_values, inputs_stacked, shape_invariants = self._create_init_values(
664 cond, body, init_values, shape_invariants=shape_invariants)
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 722 auto init_value = reduce->init_values()[i]; in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 638 GetTuple(op.init_values(), ctx), body, in ExportXlaOp()
|