Home
last modified time | relevance | path

Searched refs:remove_dynamic_dimensions (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc311 StatusOr<XlaComputation> XlaBuilder::Build(bool remove_dynamic_dimensions) { in Build() argument
313 return Build(instructions_.back().id(), remove_dynamic_dimensions); in Build()
317 bool remove_dynamic_dimensions) { in Build() argument
321 return Build(root.handle(), remove_dynamic_dimensions); in Build()
325 bool remove_dynamic_dimensions) { in Build() argument
331 if (remove_dynamic_dimensions) { in Build()
Dxla_builder.h211 StatusOr<XlaComputation> Build(bool remove_dynamic_dimensions = true);
216 bool remove_dynamic_dimensions = true);
319 StatusOr<XlaComputation> Build(int64 root_id, bool remove_dynamic_dimensions);