Searched refs:batch_dims_ (Results 1 – 1 of 1) sorted by relevance
588 int32 batch_dims_ = 0; member in tensorflow::ResourceGatherOp596 for (int idx = 0; idx < batch_dims_; ++idx) { in AddBatchOffsets()602 int64 const batch_offset = params.dim_size(batch_dims_); in AddBatchOffsets()613 OP_REQUIRES_OK(c, c->GetAttr("batch_dims", &batch_dims_)); in ResourceGatherOp()645 for (int i = 0; i < batch_dims_; ++i) { in Compute()648 for (int i = batch_dims_; i < indices.dims(); ++i) { in Compute()651 for (int i = batch_dims_ + 1; i < params.dims(); ++i) { in Compute()670 if (batch_dims_ > 0) { in Compute()682 for (int idx = 0; idx <= batch_dims_; ++idx) { in Compute()686 for (int i = batch_dims_ + 1; i < params.dims(); ++i) { in Compute()