| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_util.cc | 214 MakeInitTupleFromInitValues(const WhileUtil::LoopStateTy& init_values) { in MakeInitTupleFromInitValues()
229 const WhileUtil::LoopStateTy& init_values) { in MakeLoopStateShapeWithLayout()
247 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop()
289 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop()
|
| D | hlo_instructions.h | 663 absl::Span<HloInstruction* const> init_values() const { in init_values() function
1383 absl::Span<HloInstruction* const> init_values() const { in init_values() function
|
| D | hlo_instruction.cc | 269 auto init_values = in CreateFromProto() local
647 auto init_values = in CreateFromProto() local
1474 absl::Span<HloInstruction* const> init_values, in CreateReduce()
1494 absl::Span<HloInstruction* const> init_values, const Window& window, in CreateReduceWindow()
|
| D | shape_inference.cc | 2263 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape() local
2312 absl::Span<const Shape* const> init_values, const Window& window, in InferReduceWindowShape()
|
| D | hlo_evaluator.cc | 2419 absl::Span<const Literal* const> init_values, in GenerateReduceOutputElement()
2489 absl::InlinedVector<const Literal*, 1> init_values(num_args); in HandleReduce() local
|
| D | algebraic_simplifier.cc | 5021 auto init_values = reduce_window->init_values(); in HandleReduceWindow() local
|
| D | hlo_instructions.cc | 2377 absl::Span<HloInstruction* const> init_values, const Window& window, in HloReduceWindowInstruction()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_elemental_utils.cc | 223 ArrayRef<Value> init_values) { in createLoopAndSetInsPt()
235 ArrayRef<Value> init_values) { in createParallelAndSetInsPt()
|
| D | legalize_to_linalg.cc | 2372 ArrayRef<Value> init_values = args.drop_front(op.inputs().size()); in matchAndRewrite() local
|
| /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/mlir/tools/kernel_gen/transforms/ |
| D | bufferize.cc | 157 SmallVector<Value> init_values; in matchAndRewrite() local
|
| /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 | 2461 absl::Span<const XlaOp> init_values, in Reduce()
2526 absl::Span<const XlaOp> init_values, in ReduceWindow()
2578 absl::Span<const XlaOp> operands, absl::Span<const XlaOp> init_values, in ReduceWindowWithGeneralPadding()
2623 absl::Span<const XlaOp> operands, absl::Span<const XlaOp> init_values, in ReduceWindowInternal()
4444 absl::Span<const XlaOp> init_values, in Reduce()
4466 absl::Span<const XlaOp> init_values, in ReduceWindow()
4491 absl::Span<const XlaOp> operands, absl::Span<const XlaOp> init_values, in ReduceWindowWithGeneralPadding()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 965 llvm::SmallVector<xla::XlaOp> operands, init_values; in ExportXlaOp() local
989 llvm::SmallVector<xla::XlaOp> operands, init_values; in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 1344 std::vector<T> init_values) { in MkTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3115 absl::Span<HloInstruction* const> init_values = reduce_window->init_values(); in HandleReduceWindow() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 590 WhileBodyFnType body_fn, ArrayRef<Value> init_values, in CreateWhile32()
4578 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 | 2335 ValueRange inputs, ValueRange init_values, in build()
|