Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:while_shape (Results 1 - 4 of 4) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/
while_loop_invariant_code_motion_test.cc	61 Shape while_shape = local 67 HloInstruction::CreateParameter(0, while_shape, "param")); 83 HloInstruction::CreateParameter(0, while_shape, "init_value")); 85 while_shape, MakeAlwaysTrueComputation(while_shape, &module()), 103 Shape while_shape = local 109 HloInstruction::CreateParameter(0, while_shape, "param")); 142 HloInstruction::CreateParameter(0, while_shape, "init_value")); 144 while_shape, MakeAlwaysTrueComputation(while_shape, &module()) 176 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); local 213 Shape while_shape = local 250 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); local 289 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); local 331 Shape while_shape = local 380 Shape while_shape = local 416 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); local     [all...]
while_loop_simplifier_test.cc	423 Shape while_shape = ShapeUtil::MakeTupleShape({scalar_s32, scalar_s32}); local 428 HloInstruction::CreateParameter(0, while_shape, "param")); 438 HloInstruction::CreateParameter(0, while_shape, "init_value")); 440 while_shape, MakeAlwaysTrueComputation(while_shape, &module()),
/external/tensorflow/tensorflow/compiler/xla/tests/
params_test.cc	385 Shape while_shape = ShapeUtil::MakeTupleShape(parameter_shapes); local 390 builder.Parameter(0, while_shape, "condition_parameter"); 400 auto body_parameter = builder.Parameter(0, while_shape, "body_parameter");
while_test.cc	1213 auto while_shape = ShapeUtil::MakeTupleShape( local     [all...]

Completed in 82 milliseconds