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

  /external/tensorflow/tensorflow/compiler/xla/service/
shape_inference_test.cc 63 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_);
88 select_program_shape_ = ShapeUtil::MakeProgramShape(
90 scatter_program_shape_ = ShapeUtil::MakeProgramShape(
317 ProgramShape to_apply = ShapeUtil::MakeProgramShape(
348 ShapeUtil::MakeProgramShape({ShapeUtil::MakeShape(F32, {})}, pred_);
358 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape(
369 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape(
380 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape(
563 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_);
573 ShapeUtil::MakeProgramShape({f32_, f32_}, f32_), {0})
    [all...]
  /external/tensorflow/tensorflow/compiler/aot/
codegen_test.cc 175 compile_result.program_shape = xla::ShapeUtil::MakeProgramShape(
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 270 static ProgramShape MakeProgramShape(std::initializer_list<Shape> parameters,
shape_util_test.cc 464 ProgramShape prog = ShapeUtil::MakeProgramShape(
shape_util.cc 172 /* static */ ProgramShape ShapeUtil::MakeProgramShape(
    [all...]

Completed in 619 milliseconds