Searched refs:old_shape_ (Results 1 – 2 of 2) sorted by relevance
96 old_shape_(output_.shape()), in CollectiveAdapterImpl()118 if (old_shape_.dims() != 1) { in Flatten()119 TensorShape new_shape = TensorShape({old_shape_.num_elements()}); in Flatten()125 if (old_shape_ != output_.shape()) { in ConsumeFinalValue()126 DMAHelper::UnsafeSetShape(&output_, old_shape_); in ConsumeFinalValue()182 const TensorShape old_shape_; member in tensorflow::__anon2cb66ace0211::CollectiveAdapterImpl
2852 old_shape_ = sp_input.get_shape().as_list()2853 transposed_dense_shape = list(old_shape_) # Copy.2855 transposed_dense_shape[i] = old_shape_[p]