Home
last modified time | relevance | path

Searched defs:new_while_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier_test.cc733 Shape new_while_shape = ParseShape("(s32[1], s32[3])").ValueOrDie(); in TEST_F() local
804 Shape new_while_shape = in TEST_F() local
870 Shape new_while_shape = ParseShape("(s32[], s32[])").ValueOrDie(); in TEST_F() local
Dwhile_loop_simplifier.cc88 Shape new_while_shape = in RemoveDeadTupleIndices() local
565 Shape new_while_shape = ShapeUtil::MakeTupleShape(new_while_shape_elems); in TryRemoveConstantParams() local
1106 Shape new_while_shape = while_shape; in TryMergeInductionVariables() local
Dwhile_util.cc108 Shape new_while_shape = while_instr->shape(); in MakeInstructionsLiveIn() local