Searched refs:SetDynamicSize (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 117 parent_->SetDynamicSize(hlo, new_index, dimension, dynamic_size); in HandleGetTupleElement()
128 parent_->SetDynamicSize(hlo, index, dimension, dynamic_size); in HandleTuple()
138 parent_->SetDynamicSize(hlo, {}, broadcast_dim, dynamic_size); in HandleBroadcast()
156 parent_->SetDynamicSize(hlo, {}, dimension, dynamic_size); in HandlePad()
186 parent_->SetDynamicSize(reduce, {}, dimensions_not_reduced_count, in HandleReduce()
244 parent_->SetDynamicSize(dot, {}, iter->second, dynamic_size); in HandleDot()
255 parent_->SetDynamicSize(hlo, {}, hlo->dimensions()[dimension], in HandleTranspose()
272 parent_->SetDynamicSize(conv, {}, in HandleConvolution()
311 parent_->SetDynamicSize(hlo, index, dimension, dynamic_size); in PassThroughDynamicDimension()
340 parent_->SetDynamicSize(reshape, {}, unmodified.second, in HandleReshape()
[all …]
|
| D | dynamic_dimension_inference.h | 84 void SetDynamicSize(HloInstruction* inst, const ShapeIndex& index, int64 dim, in SetDynamicSize() function
|