Searched defs:init_values (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_util.cc | 208 HloComputation* computation, const WhileUtil::LoopStateTy& init_values) { in MakeInitTupleFromInitValues()
223 const WhileUtil::LoopStateTy& init_values) { in MakeLoopStateShapeWithLayout()
241 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop()
|
| D | hlo_instructions.h | 608 absl::Span<HloInstruction* const> init_values() const { in init_values() function
1314 absl::Span<HloInstruction* const> init_values() const { in init_values() function
|
| D | hlo_instruction.cc | 268 auto init_values = in CreateFromProto() local
534 auto init_values = in CreateFromProto() local
1297 absl::Span<HloInstruction* const> init_values, in CreateReduce()
1317 absl::Span<HloInstruction* const> init_values, const Window& window, in CreateReduceWindow()
|
| D | shape_inference.cc | 2149 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape() local
2198 absl::Span<const Shape* const> init_values, const Window& window, in InferReduceWindowShape()
|
| D | hlo_evaluator.cc | 2355 absl::Span<const Literal* const> init_values, in GenerateReduceOutputElement()
2425 absl::InlinedVector<const Literal*, 1> init_values(num_args); in HandleReduce() local
|
| D | hlo_instructions.cc | 2239 absl::Span<HloInstruction* const> init_values, const Window& window, in HloReduceWindowInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | vector_support_library.cc | 350 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeHorizontalSums()
372 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeAvxOptimizedHorizontalSums()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 519 std::vector<xla::XlaOp> init_values; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2433 absl::Span<const XlaOp> init_values, in Reduce()
2498 absl::Span<const XlaOp> init_values, in ReduceWindow()
2550 absl::Span<const XlaOp> operands, absl::Span<const XlaOp> init_values, in ReduceWindowWithGeneralPadding()
2595 absl::Span<const XlaOp> operands, absl::Span<const XlaOp> init_values, in ReduceWindowInternal()
4607 absl::Span<const XlaOp> init_values, in Reduce()
4629 absl::Span<const XlaOp> init_values, in ReduceWindow()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 1344 std::vector<T> init_values) { in MkTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 646 WhileBodyFnType body_fn, ArrayRef<Value> init_values, in CreateWhile32()
3874 Value init_values[] = {init_value, index_init_value}; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1770 ValueRange operands, ValueRange init_values, in build()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 667 std::vector<mlir::Value> init_values = load_memrefs(reduce.init_values()); in EmitUsingElementalIrEmitter() local
|