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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
while_transformer_test.cc 41 std::unique_ptr<HloComputation> BuildConditionComputation(
148 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10));
168 module_->AddEmbeddedComputation(BuildConditionComputation(1, 10));
188 module_->AddEmbeddedComputation(BuildConditionComputation(0, 5));
207 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10));
  /external/tensorflow/tensorflow/compiler/xla/service/
copy_insertion_test.cc 407 std::unique_ptr<HloComputation> BuildConditionComputation(
750 BuildConditionComputation(loop_state_shape));
    [all...]

Completed in 82 milliseconds