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