Searched refs:old_shape_ (Results 1 – 2 of 2) sorted by relevance
85 old_shape_(output_.shape()), in CollectiveAdapterImpl()107 if (old_shape_.dims() != 1) { in Flatten()108 TensorShape new_shape = TensorShape({old_shape_.num_elements()}); in Flatten()114 if (old_shape_ != output_.shape()) { in ConsumeFinalValue()115 DMAHelper::UnsafeSetShape(&output_, old_shape_); in ConsumeFinalValue()173 const TensorShape old_shape_; member in tensorflow::__anon2cb66ace0111::CollectiveAdapterImpl
2525 old_shape_ = sp_input.get_shape().as_list()2526 transposed_dense_shape = list(old_shape_) # Copy.2528 transposed_dense_shape[i] = old_shape_[p]