Searched defs:init_values (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_util.cc | 206 HloComputation* computation, const WhileUtil::LoopStateTy& init_values) { in MakeInitTupleFromInitValues()
221 const WhileUtil::LoopStateTy& init_values) { in MakeLoopStateShapeWithLayout()
239 const WhileUtil::LoopStateTy& init_values, in MakeCountedLoop()
|
| D | hlo_instructions.h | 521 absl::Span<HloInstruction* const> init_values() const { in init_values() function
|
| D | hlo_evaluator.cc | 2335 absl::Span<const Literal* const> init_values, in GenerateReduceOutputElement()
2405 absl::InlinedVector<const Literal*, 1> init_values(num_args); in HandleReduce() local
|
| D | hlo_instruction.cc | 247 auto init_values = in CreateFromProto() local
1138 absl::Span<HloInstruction* const> init_values, in CreateReduce()
|
| D | shape_inference.cc | 2106 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | vector_support_library.cc | 345 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeHorizontalSums()
367 std::vector<llvm::Value*> vectors, llvm::Value* init_values) { in ComputeAvxOptimizedHorizontalSums()
|
| /external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/ |
| D | lhlo_dialect_emitter.cc | 253 auto init_values = in HandleReduce() 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/mlir/xla/ir/ |
| D | hlo_ops.cc | 783 ValueRange operands, ValueRange init_values, in build()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 309 WhileBodyFnType body_fn, ArrayRef<Value> init_values, in CreateWhile32()
2218 Value init_values[] = {init_value, index_init_value}; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 1316 std::vector<T> init_values) { in MkTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1914 absl::Span<const XlaOp> init_values, in Reduce()
3267 absl::Span<const XlaOp> init_values, in Reduce()
|