Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops_test.cc140 reshape_ = false; in BuildNodeDef()
158 reshape_ = reshape; in BuildNodeDefWithReshape()
187 if (reshape_) { in ExecOp()
200 bool reshape_; member in tensorflow::ScopedAllocatorConcatOpTest
Dscoped_allocator_ops.cc97 OP_REQUIRES_OK(context, context->GetAttr("reshape", &reshape_)); in ScopedAllocatorConcatOp()
120 if (reshape_) { in Compute()
162 bool reshape_; member in tensorflow::ScopedAllocatorConcatOp