HomeSort by relevance Sort by last modified time
    Searched refs:starts_shape (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
liveness_util_test.cc 241 Shape starts_shape = ShapeUtil::MakeShape(S32, {1}); local
247 HloInstruction::CreateParameter(2, starts_shape, "starts"));
tuple_points_to_analysis_test.cc 609 Shape starts_shape = ShapeUtil::MakeShape(S32, {1}); local
611 ShapeUtil::MakeTupleShape({input_shape, update_shape, starts_shape});
640 HloInstruction::CreateGetTupleElement(starts_shape, tuple_param0, 2));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_instruction_fusion_test.cc 273 Shape starts_shape = ShapeUtil::MakeShape(F32, {2}); local
280 HloInstruction::CreateParameter(1, starts_shape, "starts"));

Completed in 112 milliseconds