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

  /external/tensorflow/tensorflow/compiler/xla/service/
computation_layout.h 41 const ShapeLayout& parameter_layout(int64 param_no) const { function in class:xla::ComputationLayout
layout_assignment.cc 483 &computation_layout.parameter_layout(instruction->parameter_number())
539 called_computation_layout.parameter_layout(i).shape(), instruction,
665 TF_RET_CHECK(computation_layout.parameter_layout(i).MatchesLayoutInShape(
697 condition_computation_layout.parameter_layout(0).MatchesLayoutInShape(
699 TF_RET_CHECK(body_computation_layout.parameter_layout(0).MatchesLayoutInShape(
727 TF_RET_CHECK(true_computation_layout.parameter_layout(0).MatchesLayoutInShape(
730 false_computation_layout.parameter_layout(0).MatchesLayoutInShape(
754 const ShapeLayout& parameter_layout = local
    [all...]

Completed in 195 milliseconds