Searched refs:reshape_ (Results 1 – 3 of 3) sorted by relevance
94 const Vec& reshape(int i) const { return reshape_[i]; } in reshape()114 Vec reshape_[N]; variable168 reshape_[i].push_back(elements); in BCastList()233 reshape_[i].push_back(1); in BCastList()264 reshape_[i].back() = mul_dims(reshape_[i].back(), copy[i][j]); in BCastList()274 reshape_[i].push_back(copy[i][j]); in BCastList()289 reshape_[i].push_back(1); in BCastList()295 Reverse(&reshape_[i]); in BCastList()307 ComputeBatchIndices(output_batch_size_, reshape_[i], bcast_[i], in BCastList()374 const Vec& x_reshape() const { return reshape_[0]; } in x_reshape()[all …]
98 OP_REQUIRES_OK(context, context->GetAttr("reshape", &reshape_)); in ScopedAllocatorConcatOp()121 if (reshape_) { in Compute()163 bool reshape_; member in tensorflow::ScopedAllocatorConcatOp
141 reshape_ = false; in BuildNodeDef()159 reshape_ = reshape; in BuildNodeDefWithReshape()188 if (reshape_) { in ExecOp()201 bool reshape_; member in tensorflow::ScopedAllocatorConcatOpTest