Searched refs:parameter_layouts (Results 1 – 6 of 6) sorted by relevance
85 parameter_layouts() == other.parameter_layouts(); in operator ==()90 parameter_layouts() != other.parameter_layouts(); in operator !=()
62 const std::vector<ShapeLayout>& parameter_layouts() const { in parameter_layouts() function
56 entry_computation_layout_->parameter_layouts()) { in compilation_cache_key()
408 FromC(&se_config.entry_computation_layout.parameter_layouts[i]))); in ConfigWithLayout()440 hlo_config.entry_computation_layout.parameter_layouts = in ToC()445 &hlo_config.entry_computation_layout.parameter_layouts[i]); in ToC()479 &c_config->entry_computation_layout.parameter_layouts[i]); in Free()481 delete[] c_config->entry_computation_layout.parameter_layouts; in Free()
272 XLA_Shape* parameter_layouts; member
352 .parameter_layouts(); in RunExecutable()