| /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 | 875 absl::Span<HloInstruction* const> init_values() const { in init_values() function
1687 absl::Span<HloInstruction* const> init_values() const { in init_values() function
|
| D | hlo_instruction.cc | 333 auto init_values = in CreateFromProto() local
703 auto init_values = in CreateFromProto() local
1572 absl::Span<HloInstruction* const> init_values, in CreateReduce()
1585 absl::Span<HloInstruction* const> init_values, in CreateReduce()
1616 absl::Span<HloInstruction* const> init_values, const Window& window, in CreateReduceWindow()
|
| D | shape_inference.cc | 2252 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape() local
2304 absl::Span<const Shape* const> init_values, const Window& window, in InferReduceWindowShape()
|
| D | hlo_creation_utils.cc | 447 absl::Span<HloInstruction* const> init_values, in MakeReduceHlo()
|
| D | hlo_evaluator.cc | 3660 absl::Span<const Literal* const> init_values, in GenerateReduceOutputElement()
3730 absl::InlinedVector<const Literal*, 1> init_values(num_args); in HandleReduce() local
|
| D | hlo_instructions.cc | 2538 absl::Span<HloInstruction* const> init_values, const Window& window, in HloReduceWindowInstruction()
|
| D | algebraic_simplifier.cc | 5607 auto init_values = reduce_window->init_values(); in HandleReduceWindow() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk.cc | 41 absl::Span<const XlaOp> init_values) { in GetOperandTypes()
111 absl::Span<const XlaOp> init_values, int64_t top_k, in AggregateToTopKBuilder()
158 absl::Span<const XlaOp> init_values, int64_t top_k, in ApproxTopK()
254 absl::Span<const XlaOp> init_values, int64_t top_k, in ApproxTopKFallback()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | approx_topk_op.cc | 122 absl::Span<const xla::XlaOp> init_values, int64_t top_k, in ApproxTopKFn()
140 absl::Span<const xla::XlaOp> init_values, int64_t top_k, in ApproxTopKFn()
|
| D | image_ops.cc | 543 std::vector<xla::XlaOp> init_values; in ComputeResult() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | vector_support_library.cc | 351 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeHorizontalSums()
373 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeAvxOptimizedHorizontalSums()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2588 absl::Span<const XlaOp> init_values, in Reduce()
2653 absl::Span<const XlaOp> init_values, in ReduceWindow()
2704 absl::Span<const XlaOp> operands, absl::Span<const XlaOp> init_values, in ReduceWindowWithGeneralPadding()
2749 absl::Span<const XlaOp> operands, absl::Span<const XlaOp> init_values, in ReduceWindowInternal()
4593 absl::Span<const XlaOp> init_values, in Reduce()
4615 absl::Span<const XlaOp> init_values, in ReduceWindow()
4640 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 | 1328 llvm::SmallVector<xla::XlaOp> operands, init_values; in ExportXlaOp() local
1352 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/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3480 SmallVector<Value> init_values{op.init_value()}; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 520 WhileBodyFnType body_fn, ArrayRef<Value> init_values, in CreateWhile32()
4379 Value init_values[] = {init_value, index_init_value}; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3901 absl::Span<HloInstruction* const> init_values = reduce_window->init_values(); in HandleReduceWindow() local
|