Searched refs:tmp_shape (Results 1 – 1 of 1) sorted by relevance
699 TensorShape tmp_shape = tmp.shape(); in Compute() local700 tmp_shape.RemoveDim(0); in Compute()701 OP_REQUIRES(c, tmp.CopyFrom(tmp, tmp_shape), in Compute()722 TensorShape tmp_shape = tmp.shape(); in Scatter() local723 tmp_shape.RemoveDim(0); in Scatter()724 if (!tmp.CopyFrom(tmp, tmp_shape)) { in Scatter()