HomeSort by relevance Sort by last modified time
    Searched defs:init_values (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
image_ops.cc 513 std::vector<xla::XlaOp> init_values; variable
514 init_values.reserve(4);
515 init_values.push_back(xla::ConstantR0<int32>(builder, 0)); // col_idx
516 init_values.push_back(xla::ConstantR0<int32>(builder, 0)); // num_outputs
517 init_values.push_back(iou_thresh_mask);
518 init_values.push_back(included_iou);
522 SuppressBodyFn(num_boxes), init_values,
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_instruction.cc 246 auto init_values = local
250 CreateReduce(shape, inputs, init_values,
    [all...]
shape_inference.cc 2080 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); local
    [all...]
hlo_instructions.h 474 absl::Span<HloInstruction* const> init_values() const { function in class:xla::HloReduceInstruction
    [all...]

Completed in 387 milliseconds